IsData

Checks if a value is a registered data type (Flax NNX specific).

Abstract Signature:

IsData(value)

PyTorchJAX (Core)Flax NNX

PyTorch

API: —

Strategy: Plugin (is_tensor_check)

JAX (Core)

API: —

Strategy: Macro 'isinstance({value}, (jax.Array, jax.numpy.ndarray))'

Flax NNX

API: flax.nnx.is_data

Strategy: Direct Mapping

Official Docs ↗