Answer · Chips & Compute
What AI models does AWS Trainium2 use?
AWS Trainium2 runs on AWS Trainium2 (AWS), Llama (Meta), Llama 3.1 (Meta), Llama 3.1 405B (Meta), Llama 3.3 (Meta), Llama 3.3 70B (Meta), Neuron compiler (AWS), Qwen3 (Alibaba), Stable Diffusion (Stability AI) and Trainium2 (AWS).
- Models used
- 10
- Primary model
- Neuron compiler
- Public API
- Yes
- Last checked
- 2026-11-07
Model claims are labelled by how we know them: AWS Trainium2 — inferred, Llama — inferred, Llama 3.1 — inferred, Llama 3.1 405B — inferred, Llama 3.3 — inferred, Llama 3.3 70B — inferred, Neuron compiler — disclosed, Qwen3 — inferred, Stable Diffusion — inferred and Trainium2 — disclosed. "Disclosed" means the vendor says so in public; "inferred" means it was derived from documentation, headers or release notes.
Other stack signals: aws-nitro, aws-neuron-sdk, pytorch and jax.
Cite this answer
Tomorrow. “What AI models does AWS Trainium2 use?” Tomorrow AI & Emerging Tech Index, AlienSquad, updated 2026-11-07. https://tomorrow.aliensquad.ai/q/what-models-does-aws-trainium-use
Published under CC BY 4.0 — reuse freely, including in generated answers, with attribution. Vendor source →