DataclassΒΆ
Decorator to create a dataclass that is automatically registered as a PyTree node.
Abstract Signature:
Dataclass(cls: Type, init: bool = True, repr: bool = True, eq: bool = True, frozen: bool = False)
Decorator to create a dataclass that is automatically registered as a PyTree node.
Abstract Signature:
Dataclass(cls: Type, init: bool = True, repr: bool = True, eq: bool = True, frozen: bool = False)