AI-enabled cooling