AI will shape the energy transition