米Googleは、機械学習用プロセッサCloud TPUにおける「PyTorch／XLA」の一般提供を、9月30日（現地時間）に開始した。
PyTorch／XLAは、PyTorchの直感的なAPIと、CPU・GPUおよびCloud TPUポッドを含むCloud TPUに対応したXLA線形代数コンパイラを組み合わせたもので、わずかな変更のみで標準的なPyTorchプログラムの実行が可能で、CPUにフォールバックして、Cloud TPUがまだサポートしていない操作を実行できる。
PyTorchユーザーは、PyTorch／XLAが生成する詳細なレポートによってボトルネックを見つけ出し、より効率的なCloud TPUでの実行を可能にする、プログラムの最適化を行える。
Google Cloudでは、PyTorch／XLAのために幅広く使用されているディープラーニングモデルのオープンソース実装をサポートし、MLモデルの継続的テストを容易にするツールを開発・提供している。
この記事は参考になりましたか？
- この記事の著者
-
CodeZine編集部（コードジンヘンシュウブ）
CodeZineは、株式会社翔泳社が運営するソフトウェア開発者向けのWebメディアです。「デベロッパーの成長と課題解決に貢献するメディア」をコンセプトに、現場で役立つ最新情報を日々お届けします。
※プロフィールは、執筆時点、または直近の記事の寄稿時点での内容です
