flax

Fixed typos in docs_nnx/migrating/pytorch_to_jax_flax.rst

#5479

Open