前の記事↓ manva.hatenablog.com で、Jetson Nanoに JAXをインストールし、次はFlaxをインストールできたのでやり方を書いておく。 Flaxとは、JAXベースのニューラルネット(NN)ライブラリである。知らんけど。NNライブラリと言えば、TensorFlowとPyTorchがメ…

Jetson Nano 2GB版↓にJAXをインストールできたのでやり方を書いておく。 www.switch-science.com JAXとは、GPU対応のNumpyのようなものらしい。知らんけど。なぜJetson NanoやJAXを始めたのかは後日別途語りたい。まずはJAXの公式サイト↓ github.com の手順…