Reverts eeaa805

PiperOrigin-RevId: 823735019

Author: Google-ML-Automation

jax/_src/config.py

@@ -1836,7 +1836,7 @@ def _validate_default_device(val):
 
 vjp3 = bool_state(
     name='jax_vjp3',
-    default=True,
+    default=False,
     upgrade=True,
     help='Use new backward-pass code in jax.vjp')