Skip to content
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
☆137Oct 30, 2026Updated last 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?