brainstate.transform.all_checks

brainstate.transform.all_checks = frozenset({<class 'jax._src.checkify.DivisionByZeroError'>, <class 'jax._src.checkify.FailedCheckError'>, <class 'jax._src.checkify.NaNError'>, <class 'jax._src.checkify.OOBError'>})

Build an immutable unordered collection of unique elements.