jax.experimental.checkify module#

API#

checkify(fun[, errors])

Functionalize check calls in fun, and optionally add run-time error checks.

check(pred, msg)

Check a predicate, add an error with msg if predicate is False.

check_error(error)

Raise an Exception if error represents a failure.

Error(err, code, msgs[, payload])

param err

ErrorCategory(value)

An enumeration.

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