ML4FastSim Training Pipeline