aws-neuron / torch-neuronxView on GitHub
A native PyTorch backend for AWS Trainium with support for eager execution, torch.compile, distributed training, and custom NKI kernels
☆113Oct 20, 2026Updated this week

Alternatives and similar repositories for torch-neuronx

Users that are interested in torch-neuronx are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.

Sorting:

Are these results useful?