AI HPC scheduling