Distributing training jobs