deep-learning-containers
deep-learning-containers copied to clipboard
[HuggingFace][Neuronx] Training - Optimum Neuron 0.0.25 - Neuron sdk 2.20.0 - Transformers to 4.43.2
Issue #4307
Description
This PR creates Hugginface's PyTorch DLC for training on neuron-v2 devices (Trainium).
By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license. I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.