flax

0be01f9e - [nnx] fix mnist_tutorial colab link