AI based cooling systems