Merge pull request #21077 from merrymercy:patch-1

PiperOrigin-RevId: 631409738

jax/_src/api.py

@@ -173,7 +173,7 @@ def jit(
       ``in_shardings`` or an error is raised, and the compiled computation has
       input shardings corresponding to ``in_shardings``. If not provided, the
       compiled computation's input shardings are inferred from argument
-      sharings.
+      shardings.
     out_shardings: optional, a :py:class:`Sharding` or pytree with
       :py:class:`Sharding` leaves and structure that is a tree prefix of the
       output of ``fun``. If provided, it has the same effect as applying