torch-neuronx model support Space

Use it to look up Trainium compatibility for a model and get instructions for running it yourself.

Availability

The Space answers which popular models on the Hub run on Trainium.

Pricing model

Not established

Deployment

Hugging Face Space

Privacy

Not established

Licence

Not established

• The Space tells users whether a model runs unmodified on Trainium. ^[1]

  It tells you whether a model runs unmodified on Trainium, and gives you the script to run it yourself.

• The Space provides a script to run a selected model. ^[1]

  It tells you whether a model runs unmodified on Trainium, and gives you the script to run it yourself.

• The Space reports the number of Trainium cores used, wall time, and peak device memory. ^[1]

  Each row gives the result, the number of Trainium cores the model is sharded over, the wall time of the run (compilation included) and the peak device memory.

• At the time of writing, only Transformers models are tested. ^[1]

  At the time of writing, only Transformers models are tested, and our selection is limited to text-generation and image-text-to-text models.

• At the time of writing, tests are limited to text-generation and image-text-to-text models that fit on four Trainium2 devices. ^[1]

  At the time of writing, only Transformers models are tested, and our selection is limited to text-generation and image-text-to-text models.

• A toolkit for voice agents that listen, respond, and take action Weekly brief 2026-W43
• AWS Continuum can investigate code vulnerabilities and propose fixes Daily brief 2026-10-25