PrototypeCommon
aws-neuron/torch-neuronx
A native PyTorch backend for AWS Trainium with support for eager execution, torch.compile, distributed training, and custom NKI kernels
- Owner
- aws-neuron (organization)
- Stars
- Stars: 132
- Forks
- Forks: 11
- Language
- Python
- License
- Apache-2.0
- Created
- 2025
- Topics
- awsdeep-learningmachine-learningnkipytorchtrainiumaws-neurondtensorfsdpinference
- Tier
- Prototype (100 to 249 ★) What this tier means
More repos by aws-neuron
All 4 repos by aws-neuron- aws-neuron/aws-neuron-sdkStars: 651PythonAlpha
- aws-neuron/aws-neuron-samplesStars: 163Jupyter NotebookPrototype
- aws-neuron/transformers-neuronxStars: 113PythonPrototype
Nearby in Python
All Python repos- pzs19/HarnessProStars: 132PythonPrototype
- proffitteoy/Mathematician-HumanizerStars: 132PythonPrototype
- zexuanw958-svg/dafeiyu-config-packStars: 132PythonPrototype
- jacobian/jellyrollStars: 132PythonPrototype
- moraes/tipfyStars: 132PythonPrototype
- zacharyvoase/cssminStars: 132PythonPrototype