sequince.utils.pytrees# Attributes# OPTREE_NAMESPACE Functions# pytree_dataclass(*, data_fields[, meta_fields, ...]) Register a dataclass as an Optree and, when available, JAX pytree. Module Contents# sequince.utils.pytrees.OPTREE_NAMESPACE = 'sequince'# sequince.utils.pytrees.pytree_dataclass(*, data_fields, meta_fields=(), drop_fields=())[source]# Register a dataclass as an Optree and, when available, JAX pytree.