Consulte o suporte ao Trainium antes de provisionar a máquina
O catálogo reúne resultados, versões e scripts de reprodução. Um modelo ausente ainda não significa incompatibilidade.
Verificar compatibilidade antes de provisionar uma máquina reduz uma dúvida concreta. O catálogo de suporte ao torch-neuronx, apresentado por David Corvoysier em 23 de outubro de 2026, reúne resultados de execução de modelos no AWS Trainium.
A consulta mostra versões do compilador e do Transformers, memória de dispositivo e tempo total, incluindo compilação. Também oferece o ambiente e o script usados na reprodução. Minha avaliação: esse conjunto é mais útil para investigar compatibilidade que um selo isolado de aprovação.
O significado de aprovação, porém, é limitado. O teste gera 32 tokens. Na execução sem divisão entre núcleos, compara os tokens com uma referência em CPU; na execução distribuída, usa um modelo avaliador para conferir fluência em inglês. Isso não mede o desempenho sustentado de uma aplicação com respostas longas.
Modelo ausente também não equivale a modelo incompatível: a seleção prioriza popularidade e pode simplesmente não ter alcançado aquele caso.
Serve a equipes considerando Trainium que precisam de uma configuração reproduzível. A restrição decisiva é o escopo: a consulta verifica uma execução curta, não dimensiona sua carga de produção.
Fonte: David Corvoysier, consulta de suporte ao Trainium.