Field noteOct 23, 20261 min read1 source

Does Your Model Run on AWS Trainium? Look It Up

In one minute

Hugging Face has released the torch-neuronx model support Space, a lookup tool for checking whether popular models on the Hugging Face Hub run on AWS Trainium through TorchNeuron, the native PyTorch backend in the torch-neuronx package.

What changed: Hugging Face adds a model-support lookup Space for TorchNeuron, showing which popular Transformers models run on Trainium and providing test details and runnable scripts.

For each tested model, the Space reports whether it ran unmodified, the number of Trainium cores used, wall-clock time including compilation, and peak device memory. It also exposes the tested Neuron compiler and Transformers versions, installation packages, and the run.py script used for the test. Users can select compiler and Transformers versions, and can copy matching compiled graphs from a public NEFF cache bucket.

The published results come from the public aws-neuron/torch-neuronx-model-support dataset, stored as Parquet files for pairs of compiler and library versions. Tests currently cover Transformers text-generation and image-text-to-text models, with a limit of four Trainium2 devices, or 16 cores. A PASS only records the described 32-token check; it does not establish throughput, behavior on longer generations, or compatibility beyond the tested stack and model versions.

Evidence / sources

  1. 1huggingface.co/blog/torch-neuronx-model-support
All Field Notes

Next reads

Read next

Open the graph
  1. 01topicModelsExplore the models topic hub.
  2. 02archiveField NotesOpen the latest editorial analysis.
  3. 03source ledgerSource LedgerInspect the public source evidence graph.
Who read this page?Machine requests, hidden until opened

Loading the privacy-safe route aggregate...

Open the JSON contract