TorchNeuron开源了：PyTorch迁Trainium改.to('neuron')

AWS在2026年10月20日宣布 TorchNeuron 公测：面向Trainium的开源原生PyTorch后端。官方说法是把设备切到 .to('neuron')，并兼容eager、torch.compile 和常规PyTorch API，还能接NKI写自定义算子。[1]

仓库 aws-neuron/torch-neuronx 公开页可见（星标核对于2026年10月23日约100+量级，以页面为准）。[2] 这和同周Google的TorchTPU是一条线上的两家答案：都想让你少改代码换后端。

你本地训练会试Trainium，还是继续CUDA？

[1] AWS宣布TorchNeuron公测，2026年10月20日。

[2] GitHub aws-neuron/torch-neuronx，星标核对于2026年10月23日。

#TorchNeuron #PyTorch #Trainium #AWS #开源