jax.experimental.checkify module .. automodule:: jax.experimental.checkify API .. autosummary:: :toctree: _autosummary checkify check check_error Error JaxRuntimeError user_checks nan_checks index_checks div_checks float_checks automatic_checks all_checks