jax.experimental.checkify module

API

checkify(f[, errors])	Functionalize check calls in fun, and optionally add run-time error checks.
check(pred, msg, *fmt_args, **fmt_kwargs)	Check a predicate, add an error with msg if predicate is False.
check_error(error)	Raise an Exception if error represents a failure.
Error(_pred, _code, _metadata, _payload)	param _pred:
JaxRuntimeError
user_checks	frozenset() -> empty frozenset object frozenset(iterable) -> frozenset object
nan_checks	frozenset() -> empty frozenset object frozenset(iterable) -> frozenset object
index_checks	frozenset() -> empty frozenset object frozenset(iterable) -> frozenset object
div_checks	frozenset() -> empty frozenset object frozenset(iterable) -> frozenset object
float_checks	frozenset() -> empty frozenset object frozenset(iterable) -> frozenset object
automatic_checks	frozenset() -> empty frozenset object frozenset(iterable) -> frozenset object
all_checks	frozenset() -> empty frozenset object frozenset(iterable) -> frozenset object