開発

TorchNeuronを公開ベータ提供

原題: AWS announces TorchNeuron: Native PyTorch Support for AWS Trainium (Public Beta)

重要公開ベータ

Trainiumをnative PyTorch deviceとして利用

AWSは2026年10月21日0時00分（日本時間）、TorchNeuronの公開ベータを開始しました。

TorchNeuronはAWS Trainium向けのオープンソースnative PyTorch backendです。

PyTorch codeを.to('neuron')へ切り替え、Trainium上で実行できます。

eager execution、torch.compile、標準PyTorch APIとの互換性を提供します。

Neuron Kernel Interfaceと統合し、custom operatorとperformance optimizationを追加できます。

公開物はAWS NeuronのGitHub repositoryから利用できます。

Trainium2を使うAmazon EC2 Trn2 instanceで試用できます。

研究者と開発者は、既存PyTorch codeの大規模な書き換えを減らしてTrainiumを評価できます。

今回の状態は一般提供ではなくpublic betaです。

性能と互換性の説明はAWSによるもので、独立評価ではありません。