transformers

31245775 - Add SageMakerTrainer for model paralellism (#10122)

Add SageMakerTrainer for model paralellism (#10122) * Refactor things out of main train * Store signature * Add SageMakerTrainer * Init + Copyright * Address review comments