Update the deprecation message of backend and device argument of …

…`jit` to be more actionable.

PiperOrigin-RevId: 637899890

jax/_src/pjit.py

@@ -377,11 +377,9 @@ def _parse_jit_arguments(fun: Callable, in_shardings: Any, out_shardings: Any,
 
   if backend is not None or device is not None:
     warnings.warn(
-        'backend and device argument on jit is deprecated. You can use a '
-        '`jax.sharding.Mesh` context manager or device_put the arguments '
-        'before passing them to `jit`. Please see '
-        'https://jax.readthedocs.io/en/latest/notebooks/Distributed_arrays_and_automatic_parallelization.html '
-        'for more information.', DeprecationWarning)
+        'backend and device argument on jit is deprecated. You can use'
+        ' `jax.device_put(..., jax.local_devices("cpu")[0])` on the inputs to'
+        ' the jitted function to get the same behavior.', DeprecationWarning)
     if device is not None and backend is not None:
       raise ValueError("can't specify both a device and a backend for jit, "
                        f"got {device=} and {backend=}")