AWS Trainium

Hugging Face推TorchNeuron原生支持Trainium

Hugging Face介绍TorchNeuron，这是AWS Trainium上的PyTorch原生后端，支持eager模式、torch.compile及FSDP等分布式API。Transformers、Diffusers等库可直接运行，无需再经Optimum Neuron；公开测试版将于2026年10月20日开放，可通过pip安装。