Float8E4m3fnuz

FP8 type E4M3FNUZ.

PyTorch

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

JAX (Core)

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

NumPy

API: —
Strategy: Custom / Partial

Flax NNX

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