deep-learning-containers icon indicating copy to clipboard operation
deep-learning-containers copied to clipboard

[HuggingFace][Neuronx] Training - Optimum Neuron 0.0.25 - Neuron sdk 2.20.0 - Transformers to 4.43.2

Open dacorvo opened this issue 4 months ago • 4 comments

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.

dacorvo avatar Oct 22 '24 07:10 dacorvo