PythonHub
Link
click to show
click to show
Introducing TorchNeuron: Native PyTorch backend for AWS Trainium
PyTorch and AWS introduce TorchNeuron, an open-source backend that lets developers run PyTorch models natively on AWS Trainium2 and Trainium3 accelerators with minimal code changes. Now available in public beta, it supports eager execution, torch.compile, distributed training, custom kernels through the Neuron Kernel Interface (NKI), and performance profiling using familiar PyTorch APIs.
https://pytorch.org/blog/introducing-torchneuron-native-pytorch-backend-for-aws-trainium/
1 · 124 ·