AI FOR ENERGY STORAGE OPTIMIZATION