Prepares for training on the AI lab cluster