jnp.einsum docstring missing equation arg
### Description

The documented arg list for einsum is missing the first argument.

https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.einsum.html
`jax.numpy.einsum(*operands, out=None, optimize='optimal', precision=None, _use_xeinsum=False)`
compare to
https://numpy.org/doc/stable/reference/generated/numpy.einsum.html
`numpy.einsum(subscripts, *operands, out=None, dtype=None, order='K', casting='safe', optimize=False)`
(noting the 'subscripts' for the first arg)

### What jax/jaxlib version are you using?

_No response_

### Which accelerator(s) are you using?

_No response_

### Additional system info

_No response_

### NVIDIA GPU info

_No response_
diff --git a/jax/_src/numpy/lax_numpy.py b/jax/_src/numpy/lax_numpy.py
index b94f8336c..684a280c0 100644
--- a/jax/_src/numpy/lax_numpy.py
+++ b/jax/_src/numpy/lax_numpy.py
@@ -3112,6 +3112,7 @@ the implementation.
 
 @util._wraps(np.einsum, lax_description=_EINSUM_DOC, skip_params=['out'])
 def einsum(
+    subscripts,
     *operands,
     out=None,
     optimize="optimal",
@@ -3119,6 +3120,7 @@ def einsum(
     _use_xeinsum=False,
     _dot_general=lax.dot_general,
 ):
+  operands = (subscripts, *operands)
   if out is not None:
     raise NotImplementedError("The 'out' argument to jnp.einsum is not supported.")
 
diff --git a/tests/lax_numpy_test.py b/tests/lax_numpy_test.py
index 2f7c5ca9c..f4a0a83ae 100644
--- a/tests/lax_numpy_test.py
+++ b/tests/lax_numpy_test.py
@@ -5251,7 +5251,7 @@ class NumpySignaturesTest(jtu.JaxTestCase):
     }
 
     extra_params = {
-      'einsum': ['precision'],
+      'einsum': ['subscripts', 'precision'],
       'einsum_path': ['subscripts'],
       'take_along_axis': ['mode'],
     }
['jax/_src/numpy/lax_numpy.py', 'tests/lax_numpy_test.py']