AI power management