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