wowsuch ai
Live

AWSDeveloper platform

TorchNeuron

TorchNeuron is an open-source native PyTorch backend for AWS Trainium, launched in public beta. It supports eager execution, torch.compile, standard PyTorch APIs, and integration with the Neuron Kernel Interface for custom operators and optimization.

What it does

Developers and researchers can use it to build and experiment with PyTorch models on Trainium hardware.

Availability
Public beta; available through the TorchNeuron GitHub repo for Amazon EC2 Trainium2 and Trainium3 instances
Pricing model
Not established
Deployment
Amazon EC2 Trainium2 and Trainium3 instances
Privacy
Not established
Licence
Open source
Capability
training
Modality
code
Access
SDK
Deployment
hosted
Pricing
unknown
Target user
developerresearcher

Evidence

5 claims
  • TorchNeuron launched as a public beta on October 20, 2026. [1] [2]
    Today, AWS launched the public beta of TorchNeuron, an open source native PyTorch backend for AWS Trainium.
    AWS. Quoted text is outside CC BY 4.0 and keeps its original rights.
  • TorchNeuron is a native PyTorch backend for AWS Trainium. [1] [2]
    Today, AWS launched the public beta of TorchNeuron, an open source native PyTorch backend for AWS Trainium.
    AWS. Quoted text is outside CC BY 4.0 and keeps its original rights.
  • It supports eager execution, torch.compile, and standard PyTorch APIs. [1] [2]
    With TorchNeuron, developers can enable their PyTorch code on Trainium simply by switching to .to('neuron') while maintaining full PyTorch compatibility with eager execution, torch.compile, and standard PyTorch APIs.
    AWS. Quoted text is outside CC BY 4.0 and keeps its original rights.
  • It integrates with the Neuron Kernel Interface for custom operators and performance optimization. [1]
    Additionally, TorchNeuron integrates with Neuron Kernel Interface (NKI) to enable custom operators and performance optimization for Trainium hardware.
    AWS. Quoted text is outside CC BY 4.0 and keeps its original rights.
  • TorchNeuron is available as open source through its GitHub repo for Amazon EC2 Trainium2 and Trainium3 instances. [1]
    TorchNeuron is available as open source through the TorchNeuron github repo for Amazon EC2 Trainium2 instances and Amazon EC2 Trainium3 instances.
    AWS. Quoted text is outside CC BY 4.0 and keeps its original rights.

Sources

2
1
Verified
2
Hugging FaceReputable independent
https://huggingface.co/blog/hf-torchneuron Retrieved 2026-10-28 00:51 UTC
Verified

Versions

1
VersionEffectiveChangeChanged fieldsState
12026-10-21 05:23 UTCpublicationNoneCurrent

In briefs

2

Open in the catalog

Original catalog content is licensed under CC BY 4.0, attributed to wowsuchai. Source quotations and linked content keep their original rights.