AI for supercomputing