deep-learning-containers
deep-learning-containers copied to clipboard
[HuggingFace][Neuronx] DLC for Optimum-neuron 0.0.25 - Neuron SDK 2.20.0 PyTorch 2.1.2 - Transformers 4.43.2
Overview of DLCs to update
Inference - Neuronx
Dependencies versions:
transformers: 4.43.2 torch: 2.1.2 aws-neuron-sdk: 2.20.0 optimum-neuron: 0.0.25
Training - Neuronx
Dependencies versions:
transformers: 4.43.2 torch: 2.1.2 aws-neuron-sdk: 2.20.0 optimum-neuron: 0.0.25
cc. @catalin-manciu-aws @philschmid @mmcclean-aws @Captainia @JingyaHuang