機械学習フレームワーク「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からダウンロードできる。
リリースアナウンス
「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からダウンロードできる。
(川原 龍人/びぎねっと)
[関連リンク]リリースアナウンス
その他のニュース
- 2024/10/21 機械学習フレームワーク「PyTorch 2.5」リリース
- 2024/10/18 クエリ応答の概要が記録されるようになった「BIND 9.20.3」リリース
- 2024/10/17 「Samba 4.21.1」リリース
- 2024/10/16 Microsoft、Azure上のコンテナワークロードに最適化されたLinuxディストリビューション「Azure Linux 3.0」リリース
- 2024/10/16 Zyphra、小規模言語モデル「Zamba2-7B」をリリース
Think ITメルマガ会員登録受付中
Think ITでは、技術情報が詰まったメールマガジン「Think IT Weekly」の配信サービスを提供しています。メルマガ会員登録を済ませれば、メルマガだけでなく、さまざまな限定特典を入手できるようになります。