アマゾンがAWSトレインium上でPyTorchワークロードを実行するツールキットを発表

アマゾンはTorchNeuronというオープンソースツールキットを発表した。これにより、開発者は同社のAWSトレインiumチップ上でPyTorchワークロードを実行できるようになる。PyTorchは、業界で最も人気のある人工知能開発フレームワークである。これまで、トレインiumチップを使用したAmazon Web ServicesのインスタンスでAIプロジェクトを実行するには、多くの手動作業が必要であり、開発コストが増大し、ユーザーのアクセスが困難になっていた。

TorchNeuronは、コードの大幅な修正なしにトレインiumチップでPyTorchプロジェクトを実行できるようにする。アマゾンによると、開発者はワークロードが使用すべきチップを指定するだけで済む。TorchNeuronは、最新のトレインium2およびトレインium3をサポートしている。これらのチップは、トレーニングと推論の両方の使用ケースに対応している。トレインium3は、この二つの中でより高性能なプロセッサであり、8つのプロセッサコアを持ち、それぞれのコアは異なるデータセットの計算をサポートするために調整された4つのモジュールを含んでいる。