PyTorch: バージョン2.12はデバイス非依存のtorch.accelerator.Graph、MX量子化、100倍高速なlinalg.eighをもたらします

PyTorch Foundationは2026年5月13日にフレームワークのバージョン2.12をリリースしました。このリリースは2,926件のコミットと457名のコントリビューターによるものです。マルチベンダーアクセラレーターAPIへの移行、積極的な量子化のサポート、線形代数演算の大幅な高速化をもたらし、TorchScriptを正式に削除しています。

torch.accelerator.Graphはグラフキャプチャをどのように変えますか？

torch.accelerator.Graphは、CUDA・XPU・サードパーティバックエンドを横断してグラフキャプチャとリプレイを行う新しい統一APIです。torch.xpu.XPUGraphなどのデバイス固有の実装を置き換えます。バックエンドは軽量なGraphImplInterfaceを通じて登録され、c10::Streamとtorch.Streamはバックエンド非依存のストリームチェック用の新しいis_capturing()メソッドを取得します。この実装はIntelのGuangye YuによってPR #171269と#171285を通じて提供されました。

torch.exportのMX量子化は何を可能にしますか？

torch.export.saveとtorch.export.loadがfloat8_e8m0fnuデータ型をサポートするようになりました。この変更により、MXFP4・MXFP6・MXFP8形式で積極的に圧縮されたモデルの完全なエクスポートが可能になります。コスト制約のあるエッジ環境でのLLMデプロイに不可欠です。ARMのChizkiyahu RafulによるPR #176270を通じて提供されました。

バージョン2.12はどのような高速化をもたらしますか？

linalg.eighはレガシーMAGMAバックエンドからcuSolverに移行し、バッチ処理操作にsyevj_batchedを無条件で使用します。PyTorchチームは典型的なMLワークロードのCUDA上で最大100倍の高速化を報告しており、数分かかっていた操作が数秒で完了します。もう一つの重要な最適化：torch.condのデータ依存制御フローが、CUDA 12.4の条件分岐IFノードを使用してCUDA Graphs内でキャプチャできるようになり、以前のCUDAグラフツリーへのフォールバックが解消されました。融合Adagradオプティマイザがfused=TrueサポートでAdam・AdamW・SGDに加わります。

TorchScriptの削除は何を意味しますか？

TorchScriptは2.10から非推奨となり、2.12で正式に削除されました。推奨される代替手段：モデルシリアライゼーションにtorch.export、組み込みランタイムにExecutorchを使用してください。CUDA 12.8 wheelは標準リリースマトリックスには含まれなくなりました。PyTorchは旧アーキテクチャ（Pascal・Volta）にはCUDA 12.6、BlackwellにはCUDA 13.0+を推奨しています。

Joe Spisak・Andrey Talman・Alban Desmaison出演のライブQ&Aイベントは2026年5月20日（水）太平洋標準時午前10時に予定されています。