flax
Handle pure bodies in nnx.fori_loop
#5141
Merged

Loading