Enable the C++ jax.jit fast code-path by default (again, after a bugfix).
Enable the C++ jax.jit fast code-path by default (again, after a bugfix).

PiperOrigin-RevId: 337140000

google/jax,google/jax,tensorflow/probability,google/jax,google/jax,tensorflow/probability