Training on Amazon's EMR infrastructure