Uint2

2-bit unsigned integer type.

PyTorchJAX (Core)Flax NNX

PyTorch

API: torch.uint2

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jnp.uint8

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: jax.numpy.uint2

Strategy: Direct Mapping

Official Docs ↗