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