CodeZine(コードジン)

特集ページ一覧

Python向けのオープンソース機械学習ライブラリ「PyTorch 1.12」がリリース

  • LINEで送る
  • このエントリーをはてなブックマークに追加
2022/07/04 08:00

 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行列乗算精度の変更などによって、パフォーマンスの向上も実現している。

関連リンク

  • LINEで送る
  • このエントリーをはてなブックマークに追加

あなたにオススメ

All contents copyright © 2005-2022 Shoeisha Co., Ltd. All rights reserved. ver.1.5