jax.experimental.checkify module#

API#

checkify(fun[, errors])

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

check(pred, msg, *args, **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