AI could use as much electricity as a small country within the next three years.