🧰 torch_neuronx

The items that name it 3

• Does Your Model Run on AWS Trainium? Look It Up
  articleas classification.technologies2026-10-23It is written here as torch-neuronx.
• Profiling TorchNeuron (Part 1): System-level profiling on AWS Trainium.
  writeupas frameworks and tools2026-10-21

  Package that works with upstream PyTorch to register Neuron as a native backend, using the PrivateUse1 extension point.

• TorchNeuron: running HF models on AWS Trainium is easier than ever
  articleas frameworks and tools2026-10-19It is written here as torch-neuronx.

  Package that the earlier optimum-neuron relied on, based on PyTorch XLA; also the pip name for installing TorchNeuron.