Flexible

Flexible data type base class.

JAX (Core)

API: jax.numpy.flexible
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.flexible
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.numpy.flexible
Strategy: Direct Mapping
Official Docs ↗