Pythonのための、オープンソースの機械学習ライブラリPyTorchの開発チームは、最新バージョンとなる「PyTorch 1.12」を6月28日(現地時間)にリリースした。
「PyTorch 1.12」では、バッチデータに対する機械学習の前処理用のライブラリであるTorchArrowのベータ版が収録されており、高性能CPUバックエンド、Veloxによるベクトル化された拡張可能なユーザー定義関数(UDF)、PyTorchやTensorなどとのシームレスなハンドオフおよびPyTorch DataLoaderとDataPipesへのプラグイン、Arrowインメモリ列形式による外部リーダのゼロコピーといった機能を備えた、Python DataFrameインターフェースが提供されている。
さらに、特定のパラメーターセットを使用してモジュール処理を機能的に適用できるモジュール用のFunctional API(ベータ)、Complex Convolutionsや半精度のFFT演算を可能にするComplex32(ベータ)、フォワードモードADにおけるオペレーターカバレッジの大幅な改善(ベータ)などが追加された。
そのほか、TorchscriptのデフォルトフューザのnvFuserへの更新、Ampere以降のCUDAハードウェアにおけるfloat32行列乗算精度の変更などによって、パフォーマンスの向上も実現している。
- 関連リンク
この記事は参考になりましたか?
- この記事の著者
-
CodeZine編集部(コードジンヘンシュウブ)
CodeZineは、株式会社翔泳社が運営するソフトウェア開発者向けのWebメディアです。「デベロッパーの成長と課題解決に貢献するメディア」をコンセプトに、現場で役立つ最新情報を日々お届けします。
※プロフィールは、執筆時点、または直近の記事の寄稿時点での内容です