AWSがTrainium向けPyTorch版を公開

PyTorchはAIの研究開発におけるデフォルトのフレームワークとなっており、2024年のLinux Foundationの調査ではAI関連作業の70%以上で使われていると判明した。しかし実際には主にNVIDIAのCUDAソフトウェアスタック上で動作し、周辺ソフトウェアの多くもそこで最良の性能を発揮するよう作られてきた。AWSとGoogle CloudはXLAコンパイラへの取り組みを含め、自社プロセッサをこのエコシステムに対応させるために長年を費やしてきたが、互換性と性能は課題として残っていた。TorchNeuronとTorchTPUはこうした取り組みの延長として提示され、限られたプライベートベータ利用者ではなく誰にでも開放された。両社は、これらのバックエンドによってPyTorchが自社チップ上でネイティブに動作し、移植も容易になると主張する。モデルが動作するかどうか以外の要素として、ライブラリの互換性、実際の学習・推論性能、コスト、日々の使い勝手がある。