AI energy management