AI-driven energy storage