Share story
AI Infrastructure

PyTorch announces native Trainium backend with public beta due October 20

PyTorch announces native Trainium backend with public beta due October 20 Image: Primary
PyTorch announced TorchNeuron, an open-source backend intended to let developers run standard PyTorch models and tools on AWS Trainium2 and Trainium3 chips. Its public beta is scheduled for October 20 under the Apache 2.0 license. The backend supports immediate execution for research and debugging, compiled execution for production, distributed training APIs and custom hardware kernels. It also integrates PyTorch's profiler with Neuron Explorer to trace performance bottlenecks from software to hardware. Code using PyTorch's device-agnostic accelerator API can run unmodified, PyTorch says. Adaptive eager execution remains experimental, while TorchAO and FP8 support are in private beta.
Sources
PyTorch
In this story
Amazon
Published by Tech & Business, a media brand covering technology and business. This story was sourced from PyTorch and reviewed by the T&B editorial agent team.
Back to Newswire
Keep reading
Full wire
Capital AI
Capital AI

Nevis raises $70M with Schwab joining wealth software investors

Nevis raised $70M in a Series B led by EQT Ventures, bringing its total funding to $110M, the New York wealth management software company announced Monday. Charles Schwab and QuantumLight joined existing investors Sequoia Capital,...

The Next Web
Robotics Capital
Robotics Capital

Bedrock Robotics raises $600M at a $4.1B valuation

Bedrock Robotics raised a $600M Series C at a $4.1B valuation, Bloomberg reported. The startup makes driverless excavators and other autonomous construction machines. The financing adds capital to a company developing autonomous ...

Techmeme
Capital
Capital

Matiks raises $10 million to expand puzzle games and game creation

India-based Matiks said Monday it raised $10 million in pre-Series A funding led by General Catalyst, bringing its total funding to over $13.5 million. The startup plans to expand into new markets and scale Jelo, its experimental ...

TechCrunch
AI Products
AI Products

AWS makes GPT-6.1 Terra generally available on Bedrock

AWS announced general availability of OpenAI's GPT-6.1 Terra on Amazon Bedrock, giving customers access through Bedrock APIs and configurations for ChatGPT Work and Codex. The model accepts text and images and supports up to 1 mil...

Recent Announcements
AI
AI

OpenAI releases GPT-6.1 Terra at $1 per million input tokens

OpenAI has released GPT-6.1 Terra, priced at $1 per million input tokens and $5 per million output tokens. The company describes the model as supporting computer use at scale and detailed visual reasoning, and claims clearer, more...

Techmeme