flax

aab0c99f - Add explanation of nnx.view functions