jax.enable\_checks ================== .. currentmodule:: jax .. autodata:: enable_checks