機械学習フレームワークPyTorchの最新版となる「PyTorch 2.0」リリース

The PyTorch Foundationは現地時間2023年3月15日、機械学習フレームワーク「PyTorch」の最新版となるPyTorch 2.0のリリースを公式サイトで発表した。ソースコードやリリースノートはGitHubに設置してある。

深層学習モデルTransformerを実装するPyTorch Transformer APIを高速化するアーキテクチャーAccelerated PyTorch 2 Transformers(安定版)、ユーザーの環境に応じたモデル構築のための関数torch.compile(ベータ版)の導入などパフォーマンスを向上させる機能が数多く搭載されるPyTorch 2.0は、v1.13.1以降の428人のコントリビューターによる4,541件を超えるコミットで正式版へと至っている。公式サイトにはPyTorch 2.0の新機能解説から環境ごとのインストールなどが詳細に記された「Get Started」も用意されている。