Amazon、PyTorchコードをTrainiumで動かす「TorchNeuron」公開へ

Amazonは現地時間19日、公式ブログで、PyTorchのコードを変更せずに同社のAIチップ「Trainium」上で直接実行できるツール「TorchNeuron」を公開すると発表した。開発者は設定を1カ所変えるだけで、PyTorchのコードを新しいチップへ移せるようになる。

TorchNeuronはPyTorch Conference North Americaで20日にオープンソースとして公開される。対応するチップは「Trainium2」と「Trainium3」で、開発者は既存のテストやデバッグ、スケーリングのツールをそのまま使い続けられるという。

Amazonによると、Hugging FaceはAI開発ツールでTrainiumをデフォルトでサポートしている。音声AI企業のCartesiaは、PyTorchコードをほとんど変更せずにTrainium上で推論処理を稼働させている。統合失調症などの神経精神疾患の治療を研究するRiver Neuroscience Instituteは、創薬にTorchNeuronを活用している。

これまでPyTorchを使う開発者が別のチップへ移行するには、コードの書き換えと検証に数カ月かかっていた。AmazonはPyTorchコミュニティとも協力し、PyTorchが特定のチップに縛られないようコード改善を進めている。

Amazonは、スタートアップを含む多くの企業にとって最大のコストはチップではなく開発者の時間だと強調している。チップ間の移行を容易にすることで、チップ企業間の競争が促され、開発コストの低下につながるという。（digitaltoday.co.kr）