機械学習フレームワーク「PyTorch 2.6」リリース

1月30日 13:24

 PyTorch FoundationFacebookは1月29日(現地時間)、機械学習フレームワーク「PyTorch 2.6」をリリースした。

 「PyTorch」は、ベータ版としてではあるがtorch.compile()においてPython 3.12を使用したモデル最適化への対応や、新しいハイレベルPythonカスタム演算子APIの搭載、CPUのAOTインダクタのフリーズフラグの制御、また、AWS Graviton(aarch64-linux)プロセッサ向けのtorch.compile の最適化パフォーマンスの向上やTorchInductor におけるBF16シンボリック最適化など、いくつかの機能強化、変更などが施されている。

 「PyTouch 2.6」では、eagerモードとInductorモードの両方でx86 CPUのFP16 がサポートされるようになったほか、プロトタイプ形式でIntelグラフィックスのユーザエクスペリエンスが向上するなど、いくつかの機能強化が加わっている。

 「PyTouch 2.6」は、GitHubからダウンロードできる。

(川原 龍人/びぎねっと)

[関連リンク]
リリースアナウンス

この記事をシェアしてください

人気記事トップ10

人気記事ランキングをもっと見る