Hugging Face представила TorchNeuron, плагин для PyTorch, который обеспечивает нативную поддержку AWS Trainium. В отличие от предыдущего решения Optimum Neuron, требующего использования специфических классов и методов компиляции, TorchNeuron позволяет работать с Trainium как с обычным устройством PyTorch. Это дает возможность использовать стандартные библиотеки, такие как Transformers, Diffusers, Accelerate, PEFT и TRL, без необходимости в специализированных форках кода. Новый бэкенд поддерживает eager mode, torch.compile и распределенные API, включая FSDP и тензорный параллелизм. По заявлению компании, это значительно расширяет совместимость: теперь «из коробки» поддерживается 89% из 100 самых популярных архитектур для генерации текста, тогда как ранее этот показатель составлял 46%. Публичная бета-версия TorchNeuron станет доступна 20 октября 2026 года через установку пакета torch-neuronx.