CodeZine(コードジン)

特集ページ一覧

Google、Cloud TPUでのPyTorch/XLAの一般提供を開始

  • ブックマーク
  • LINEで送る
  • このエントリーをはてなブックマークに追加
2020/09/30 18:40

 米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モデルの継続的テストを容易にするツールを開発・提供している。

関連リンク

  • ブックマーク
  • LINEで送る
  • このエントリーをはてなブックマークに追加
All contents copyright © 2005-2020 Shoeisha Co., Ltd. All rights reserved. ver.1.5