AI-driven energy