flax

e061dff5 - Fixed docstring visibility for nnx.eval_shape

Fixed docstring visibility for nnx.eval_shape Now the docstring is not rendered: https://flax.readthedocs.io/en/latest/api_reference/flax.nnx/transforms.html#flax.nnx.eval_shape This PR should make the docstring visible