Is there some tutorials about flax nnx api with numpyro？

[sejabs]

I found offical tutorials help using old flax linen api, not new nnx. The nnx api use separatable Graphdef and States to implement flexible and pytorch-like neural netwok module.

[fehiepsi]

There isn’t. Please create a feature request.