rotative
huggingface blog · October 19, 2026

Hugging Face brings native PyTorch support to AWS Trainium

hugging face releases torchneuron, a native pytorch backend for aws trainium, making it easier to run hf models on the ai chip. this affects developers and researchers working with hf libraries.

hugging face has released torchneuron, a native pytorch backend for aws trainium, which enables support for eager mode, torch.compile, and native distributed apis like fsdp and tp. this plugin integrates directly into pytorch and exposes neuron as a native device. hf libraries such as transformers, diffusers, accelerate, peft, and trl now run directly on trainium, with 89% of the top 100 most-used text-generation model architectures supported out of the box. developers and researchers can install torchneuron with pip install torch-neuronx starting october 20, 2026.

read at source ↗

← rotative