AI tools for optimizing energy grid management