brainstate.transform.nan_checks# brainstate.transform.nan_checks = frozenset({<class 'jax._src.checkify.NaNError'>})# Build an immutable unordered collection of unique elements.
