AWSがTorchNeuronをパブリックベータで公開、AWS Trainium向けにネイティブなPyTorchサポートを追加

AWSは、AIチップ「AWS Trainium」向けに最適化されたオープンソースのネイティブPyTorchバックエンド「TorchNeuron」をパブリックベータとして公開しました。Trainiumは、深層学習モデルのトレーニングおよび推論向けに設計されたAWS独自のポートフォリオで、高いパフォーマンスとコスト効率の提供を目的としています。今回のTorchNeuronの導入により、AI研究者やエンジニアは既存のPyTorch環境に近い操作感で、Trainiumのリソースを活用した実験やモデル構築が行えるようになります。

本リリースの主な目的は、深層学習における計算資源のコスト削減とイノベーションの加速です。TorchNeuronがネイティブなバックエンドとして動作することで、PyTorchベースの既存ワークフローを大きく変更することなく、AWS専用ハードウェアの演算能力を直接活用できる設計となっています。対象ユーザーはスタートアップから大企業まで幅広く想定されており、各ニーズに合わせた価格モデルやサポートリソースも提供されます。