flax
Add `nnx.fori_loop`
#4353
Merged

Loading