<p>Today, AWS launched the public beta of TorchNeuron, an open source native PyTorch backend for AWS Trainium. Trainium is a portfolio of AI chips designed to deliver high performance and cost efficiency for deep learning AI training and inference workloads. TorchNeuron enables AI scientists and researchers to experiment and build models with PyTorch natively on Trainium.</p>
<p>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. Additionally, TorchNeuron integrates with Neuron Kernel Interface (NKI) to enable custom operators and performance optimization for Trainium hardware.</p>
<p>TorchNeuron is available as open source through the <a href="https://github.com/aws-neuron/torch-neuronx">TorchNeuron github repo</a> for Amazon EC2 Trainium2 instances and Amazon EC2 Trainium3 instances.</p>
<p>To learn more about TorchNeuron and AWS Trainium accelerators, visit the <a href="https://aws.amazon.com/ai/machine-learning/neuron/">AWS Neuron page</a>.</p>
AWS announces TorchNeuron: Native PyTorch Support for AWS Trainium (Public Beta)
From the official release
This is a short excerpt. Read the full announcement on the official source.
Continue on aws.amazon.com → (opens in a new window)Related stories
Horizon Context: Put Business Knowledge to Work for AI
Snowflake Blog
A business leader asks an AI agent: Why did customer retention decline this quarter? Before it can deliver a useful answer, the agent needs to know how your business defines retention, which customer data to use and how that data connects to the reports your teams rely on. A flue
NVIDIA Cosmos3-H-Dreams: Bringing Clinical Surgery to Life Through Generative Simulation
Hugging Face Blog
a]:hidden" NVIDIA Cosmos3 H Dreams: Bringing Clinical Surgery to Life Through Generative Simulation Enterprise + Article Published October 20, 2026 Upvote 2 Lukas Zbinden lzbinden Follow nvidia Javier Gamazo javirk1 Follow nvidia Nigel Nelson nigeln Follow nvidia Mostafa Toloui mtoloui Follow nvidia Sean Huver shuve...
MemCurrent: Designing Persistent Memory for Production Agents
Databricks Blog
AI tools have made it easy to vibe code agents. Building and maintaining an enterprise...
Introducing Agent Memory on Databricks
Databricks Blog
Today we're introducing agent memory on Agent Bricks, Databricks' agent developer...
Accelerating Mamba-2 with High-Performance Forward Pass Kernels in NVIDIA CUDA Tile
NVIDIA Developer Blog
NVIDIA CUDA Tile (cuTile) is a Python-based DSL and parallel programming model for NVIDIA GPUs that automatically leverages advanced hardware capabilities to... NVIDIA CUDA Tile (cuTile) is a Python-based DSL and parallel programming model for NVIDIA GPUs that automatically lever
Introducing Amazon ECS Gateway: A fully-managed ingress solution for ECS services
AWS News BlogChanny Yun (윤석찬)
When platform teams manage containerized applications at scale on Amazon Elastic Container Service (Amazon ECS) , configuring ingress networking for each service often becomes a bottleneck. For every new service, teams must provision and connect Application Load Balancers (ALBs)