AWS Trainium ve Inferentia çipleri için geliştirilen TorchNeuron, 20 Ekim 2026 itibarıyla `pip install torch-neuronx` komutuyla genel beta olarak erişilebilir hale geldi.
Eklenti, PyTorch'a `neuron` adında yerel bir cihaz arka ucu ekleyerek eager execution, `torch.compile` ve PyTorch'ün kendi dağıtık primitiflerini (DTensor, FSDP2) Trainium üzerinde doğrudan çalıştırıyor.
Hugging Face ekibi, bu değişiklikle Optimum Neuron üzerinden gitmek yerine Transformers, Diffusers, Accelerate, PEFT ve TRL kütüphanelerinin standart sınıfları ve betikleriyle Trainium kullanımını öneriyor.