ブックマーク / kiszk.github.io (1)

  • JuliaからCloud TPUを使う論文の、ざっくりまとめ | Kazuaki Ishizaki – Researcher and contributor for system software

    この記事は? この記事は、TensorFlow Advent Calendar 2018の17日目の記事です。 この記事の内容は? Arxivに投稿された論文、Automatic Full Compilation of JULIA Programs and ML Models to Cloud TPUsのまとめです。特に、TPUのためのコードを生成するためにはどうすればよいか、ということについてまとめます。一言でまとめれば、 です。この論文に書かれている内容は、XLA.jiで試すことができるようです。ただし、残念ながら論文に書かれている部分はコンパイル済みバイナリとして提供されていて、実装を見ることはできません。 とにかく、JuliaのコードをTPUで動かしてみたい人向け 論より証拠、という方はXLA.jiとして公開が実装されているので、こちらを実行してみてください。詳しい使い方は、XLA

  • 1