オープンソース

機械学習フレームワーク「PyTorch 1.8」リリース

 Facebookは3月8日(現地時間)、機械学習フレームワーク「PyTorch 1.8」をリリースした。

 「PyTorch」は、Python向けのテンソル操作やディープラーニング処理を集めたフレームワーク。NumPy、SciPy、Cythonなどを組み合わせることも可能。JITコンパイラを持ち、Torch Scriptを利用して既存のコードを利用できる。既存のPyTorchフレームワークとCaffe2のプロダクション機能を組み合わせることで、シームレスな機械学習の機能を提供する。

 「PyTorch 1.8」では、torch.fxによるpython to python関数型変換のサポート、NCCLの信頼性強化およびRPCプロファイリングのサポートなど、いくつかの機能強化、APIの強化など、複数の改善が施されている。

 「PyTouch 1.8」は、GitHubからダウンロードできる。

(川原 龍人/びぎねっと)

[関連リンク]
Blogによる記事