IT之家 10月20日情報、現地時間10月19日、アマゾンはオープンソースツール「TorchNeuron」を発表しました。このツールにより、開発者が作成したPyTorchコードを、アマゾンの自社開発チップ「Trainium」上で設定を1回変更するだけで実行でき、コードの書き直しは不要です。
IT之家の情報によると、TorchNeuronは、現地時間10月20日にカリフォルニア州サンノゼで開催される北米PyTorchカンファレンスで、オープンソースライセンスにより正式にリリースされます。
PyTorchはMetaが主導するオープンソースフレームワークで、大多数のAI開発者に採用されています。NVIDIAは10年以上にわたりCUDAツールを開発し、PyTorchコードをNVIDIAのチップ上で効率的に実行可能にしています。
報告によると、アマゾンのエンジニアは、従来のチップとTrainiumチップの両方で同じAIトレーニングスクリプトを実行しました。唯一の違いは、コード内で使用するチップを指定する設定パラメータのみでした。結果として、AIモデルは両環境でまったく同じ内容を学習し、学習速度も同じでした。これは、開発者がAIモデルのテスト、デバッグ、拡張に使用していたツールを、一切変更せずにTrainium上で直接使用できることを意味します。
アマゾンは、TorchNeuronがAI業界に長く欠けていた重要な要素、すなわち真に実行可能な選択肢をもたらしたと述べています。開発者が追加のコストを伴うことなく異なるチップ間で自由に切り替えられるようになると、チップメーカーは開発者の支持を得るためにより競争力のある製品を提供しなければならず、その結果、開発者はコスト削減とイテレーション速度の向上を実現できます。
また、アマゾンはPyTorchのアップストリーム開発に積極的に参加し、PyTorchコミュニティと協力して、このフレームワークを特定のチップに結びつけるコードを削除しています。その目標は、PyTorchをTrainiumユーザーに限定せず、開発者コミュニティ全体にとってより「ハードウェア非依存」なものにすることです。