Float8E5m2fnuz

FP8 type E5M2FNUZ.

PyTorch

API: torch.float8_e5m2fnuz
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jnp.float8_e5m2fnuz
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: —
Strategy: Custom / Partial

Flax NNX

API: jnp.float8_e5m2fnuz
Strategy: Direct Mapping
Official Docs ↗