AWS が TorchNeuron を発表: AWS Trainium のネイティブ PyTorch サポート (パブリックベータ版)

投稿日: 2026年10月20日

本日、AWS は AWS Trainium 向けのオープンソースのネイティブ PyTorch バックエンドである TorchNeuron のパブリックベータ版をリリースしました。Trainium は、深層学習 AI のトレーニングおよび推論ワークロードに対して高いパフォーマンスとコスト効率を提供するように設計された AI チップのポートフォリオです。TorchNeuron により、AI サイエンティストや研究者は Trainium 上でネイティブに PyTorch を使用してモデルの実験や構築を行うことができます。

TorchNeuron を使用すると、開発者は eager execution、torch.compile、および標準の PyTorch API との完全な互換性を維持しながら、単に .to('neuron') に切り替えるだけで Trainium 上で PyTorch コードを有効にすることができます。さらに、TorchNeuron は Neuron Kernel Interface (NKI) と統合されており、カスタムオペレーターの利用や Trainium ハードウェア向けのパフォーマンス最適化を可能にします。

TorchNeuron は、Amazon EC2 Trainium2 インスタンスおよび Amazon EC2 Trainium3 インスタンス向けに、TorchNeuron の GitHub リポジトリを通じてオープンソースとして提供されています。

TorchNeuron および AWS Trainium アクセラレーターの詳細については、AWS Neuron のページをご覧ください。