AI for the electric grid