jax.experimental.checkify.nan_checks# jax.experimental.checkify.nan_checks = frozenset({<class 'jax._src.checkify.NaNError'>})# frozenset() -> empty frozenset object frozenset(iterable) -> frozenset object 构建一个不可变的无序唯一元素集合。