English translation is being prepared.
On October 19, 2026, Amazon launched the TorchNeuron open-source toolkit, making AWS Trainium a standard PyTorch device. According to the PyTorch blog, this native backend lets model code run simply by changing it to .to('neuron'); it is built on the PrivateUse1 and torch.accelerator extension points, supports eager, adaptive eager, and torch.compile, and covers distributed APIs such as DDP, FSDP2, and tensor parallelism as well as NKI custom operators; the company says it has conducted coverage testing on 100,000 Hugging Face models and plans to open a public beta on October 20 under the Apache 2.0 license at github.com/aws-neuron/torch-neuronx. Hugging Face released an introduction the same day stating that the plugin supports eager mode, torch.compile, and distributed interfaces such as FSDP2 and DTensor, that the public beta begins October 20 and can be installed via pip install torch-neuronx, and that Transformers, Diffusers, Accelerate, PEFT, and TRL can run directly on Trainium without going through Optimum Neuron; in text generation, the first 1
Matched fieldsTitle
View story:Amazon launches the TorchNeuron open-source toolkit