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は、株式会社翔泳社が運営する開発者のための情報メディアです。日々の開発に取り組むエンジニアやテクノロジーを学びたい方に向けて、プログラミングやAI活用、開発ツール、エンジニアの学びとキャリアに関する記事をお届けしています。
※プロフィールは、執筆時点、または直近の記事の寄稿時点での内容です
この記事は参考になりましたか？
この記事をシェア
