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

2024年10月21日(月)

　PyTorch Foundationは10月17日(現地時間)、機械学習フレームワーク「PyTorch 2.5」をリリースした。

　「PyTorch」は、Python向けのテンソル操作やディープラーニング処理などを集めたフレームワーク。NumPy、SciPy、Cythonなどを組み合わせることも可能。JITコンパイラを持ち、Torch Scriptを利用して既存のコードを利用できる。

　「PyTouch 2.5」では、SDPA向けの新しいCuDNNバックエンドが搭載されたほか、H100以降のGPUでSDPAを使用するユーザ向けにデフォルトで高速化が有効になった。また、torch.compileのリージョンコンパイルによってユーザは繰り返しのnn.Module (LLMのトランスフォーマーレイヤーなど) を再コンパイルせずにコンパイルできるため、torch.compileのコールドスタートアップ時間を短縮できるようになるなど、いくつかの新機能が追加されている。

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

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

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