Elixirでディープラーニング②：Nxのdefn＋XLAでGPUを動か…せなかった(T_T) - Qiita

Elixir Digitalization Implementors／fukuoka.ex／kokura.exのpiacereです ご覧いただいて、ありがとうございます 引き続き、Elixirの2021年を占うポジションであるニューラルネットワーク／ディープラーニング用ライブラリ「Nx」についてです 前回は、Nxの導入を行ったので、今回は、Nxで新しく入った構文「defn」で行列処理専用関数を作ったり、Nxで提供されるTensorFlow用コンパイラ「XLA（Accelerated Linear Algebra）」バインディングを通じたGPU駆動を試していきます なおNxは、昨日（2021年2月18日）出たばかりのライブラリで、hexdocも未だ無いような状態ですが、ソースコード中のdoctestを参考にしながら実験していきます Nxソースコード https://github.com/el