Amazon、PyTorchコードをTrainiumチップでそのまま動かすTorchNeuronをオープンソース公開

要約

AWSとPyTorchプロジェクトは、AWS Trainiumチップを標準のPyTorchデバイスとして扱えるオープンソースバックエンド「TorchNeuron」を発表した。デバイス指定をneuronに変える1行だけで、既存のPyTorchモデルやライブラリがTrainium2/Trainium3でそのまま動き、eager実行、torch.compile、カスタムカーネル用インターフェースに対応する。10月20日からApache 2.0ライセンスの公開ベータとしてGitHubで提供される。研究者が使い慣れたツールのままチップの選択肢を広げるのが狙いだという。

仕事との関連

AI研究の事実上の標準であるPyTorchの生態系に代替チップが滑らかに接続されれば、実験コストとインフラ選択の幅が変わる。NVIDIA中心のインフラ市場で、オープンソースソフトウェアによる競争構図が一段と強まっている。

韓国語の原文を翻訳した記事です。 要約には翻訳と編集を含みます。コメントは私たちの解釈であり、将来予測は出典の見解です。