Home icon

AWS announces TorchNeuron: Native PyTorch Support for AWS Trainium (Public Beta)

News

AWS launched the public beta of TorchNeuron, an open source native PyTorch backend for AWS Trainium AI chips designed for deep learning training and inference workloads.

  • Enable PyTorch code on Trainium by switching to .to('neuron') with full PyTorch compatibility
  • Supports eager execution, torch.compile, and standard PyTorch APIs
  • Integrates with Neuron Kernel Interface (NKI) for custom operators and performance optimization
  • Available for Amazon EC2 Trainium2 and Trainium3 instances
  • Open source through TorchNeuron GitHub repository

TorchNeuron simplifies PyTorch model development on AWS Trainium hardware while maintaining native PyTorch workflows.

Go to article

The AWS News Feed is currently looking for gold sponsors. If you want to support the AWS community and reach a large audience of AWS professionals, consider sponsoring the AWS News Feed.

Related articles

Oct 21
2026
AWS Neuron 2.33 adds native PyTorch support, Neuron Observer, and other new features
Oct 26
2026
AWS Weekly Roundup: Amazon Polly Generative engine, TorchNeuron for Trainium, and more (October 26, 2026)
Jul 30
2026
Stop hand-porting kernels: migrate PyTorch and Triton kernels to AWS Trainium with an AWS Transform custom agent
Dec 23
2024
AWS Neuron introduces support for Trainium2 and NxD Inference

The AWS News Feed is currently looking for silver sponsors. If you want to support the AWS community and reach a large audience of AWS professionals, consider sponsoring the AWS News Feed.