Flax

概要

こちらは Flax による画像分類のコードです。前回 Haiku & Optax の組み合わせで、コードを書いてみましたが、今回はそのコードを Flax 向けに改造してみたいと思います。

基本的には Neura ...