AI-driven cooling system