Add cross-reference to lax.GatherScatterMode from jax.numpy.ndarray.at documentation.
Add cross-reference to lax.GatherScatterMode from jax.numpy.ndarray.at documentation.

google/jax,google/jax,google/jax,google/jax