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からダウンロードできる。
(川原 龍人/びぎねっと)
[関連リンク]
リリースアナウンス
この記事をシェアしてください
この記事の筆者
関連記事
Think ITでは、技術情報が詰まったメールマガジン「Think IT Weekly」の配信サービスを提供しています。メルマガ会員登録を済ませれば、メルマガだけでなく、さまざまな限定特典を入手できるようになります。
