AI-driven surrogate models can accelerate simulations for thermal estimation in chip design