---
type: "post"
id: "does-your-model-run-on-aws-trainium-look-014c50"
slug: "does-your-model-run-on-aws-trainium-look-014c50"
title: "Does Your Model Run on AWS Trainium? Look It Up"
description: "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."
retrieval_nugget: "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."
published_at: "2026-10-23"
updated_at: "2026-10-23"
record_date: "2026-10-23"
date_kind: "published_at"
topics: ["models"]
entities: ["Hugging Face","AWS Trainium","TorchNeuron","torch-neuronx","Transformers","Trainium2"]
related_patterns: []
novelty_label: "new_fact"
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."
prior_refs: ["memory:6fd99df8-1e84-87ab-a47a-3dd0758dc899","memory:e4c94c61-7406-8b23-a98d-a000597352f2","memory:b442cbe8-53c1-8af1-a5cb-9b69a67f000b"]
source_url: "https://huggingface.co/blog/torch-neuronx-model-support"
source_urls: ["https://huggingface.co/blog/torch-neuronx-model-support"]
schema_version: "newruntime-agent-readable-v0.2"
stable_id: "post:does-your-model-run-on-aws-trainium-look-014c50"
status: "published"
visuals: []
editor: {"role":"editor","name":"Andrey Reshetnikov","profile":"https://newruntime.com/owner-profile.md","profile_json":"https://newruntime.com/owner-profile.json","same_as":["https://www.linkedin.com/in/reshetnikov1/","https://t.me/lapetuse"]}
routes: {"html":"https://newruntime.com/posts/does-your-model-run-on-aws-trainium-look-014c50/","markdown":"https://newruntime.com/posts/does-your-model-run-on-aws-trainium-look-014c50.md","json":"https://newruntime.com/posts/does-your-model-run-on-aws-trainium-look-014c50.json"}
---

# Does Your Model Run on AWS Trainium? Look It Up

## Retrieval answer

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.

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.

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.

Editor: [Andrey Reshetnikov](https://newruntime.com/owner-profile.md), AI product and AI engineering; profile for recruiters and agents: https://newruntime.com/owner-profile.json

## Continue in the graph

Next records for an agent, linked to their Markdown twins. Replace `.md` with `.json` for structured fields, including `next_reads`.

- topic hub: [Models](https://newruntime.com/topics/models.md): Explore the models topic hub.
- collection: [Field Notes](https://newruntime.com/posts.json): Open the latest editorial analysis.
- collection: [Source Ledger](https://newruntime.com/source-ledger.json): Inspect the public source evidence graph.
- whole graph: [discovery graph](https://newruntime.com/discovery-graph.json) (typed nodes and edges) and [site index](https://newruntime.com/site-index.md)