AI-driven cooling