brainstate.transform.float_checks

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

Build an immutable unordered collection of unique elements.