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

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

　「PyTouch 2.4」では、「Accelerated PT2 Transformer」が搭載され、scaled dot product attention向けのアーキテクチャを利用したトレーニング・推論をサポートした。また、メインAPIとして、「torch.compile」のベータリリースが採用されるなど、いくつかの新機能が追加されている。

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

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

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