AI for energy transition