Data

Annotates an attribute as pytree data.

Abstract Signature:

Data(value)

PyTorchJAX (Core)NumPyFlax NNX

PyTorch

API: torch.nn.Parameter

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Custom / Partial

NumPy

API: numpy.f2py.symbolic.Expr.data

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: flax.nnx.data

Strategy: Direct Mapping

Official Docs ↗