Amazonは10月19日（現地時間）、PyTorchのコードを修正せずに自社AIチップ「Trainium」で実行できる「TorchNeuron」をオープンソース化すると、公式ブログで発表した。10月20日に開催される「PyTorch Conference North America」で公開する。
Amazonによると、PyTorchはAI研究者や開発者の85％が利用している主要な開発ツール。これまで開発者は、新たなチップへ移行する際にコードの書き換えや検証に数カ月を要することがあったが、TorchNeuronを使えば設定変更だけで移行できるという。
既存のテスト、デバッグ、拡張ツールも引き続き利用できるとしている。
Amazonは、スタートアップをはじめ多くの企業にとって、最大のコストはチップそのものではなく、開発者が新しいハードウェア向けにソフトウェアを調整する時間だと強調した。その分だけ製品改良が遅れるためとしている。
また、開発者がチップを容易に切り替えられるようになれば、半導体メーカー間の競争が進み、開発コストの低下にもつながると説明した。
Amazonはあわせて、PyTorchが特定のチップに依存しないよう、PyTorchコミュニティと連携してコードベースの改善も進めているという。
導入事例も紹介した。音声AI企業のCartesiaは、PyTorchコードをほとんど修正せずにTrainium上で推論を実行している。統合失調症など神経・精神疾患の治療法を研究するRiver Brain Development Instituteも、新薬開発にTorchNeuronを活用している。Hugging Faceは、自社のAI開発ツールでTrainiumを標準サポートする。
TorchNeuronは、Trainium2およびTrainium3をサポートする。
Chi-gyu Hwang
delight@d-today.co.kr