Float8E5m2

FP8 type E5M2.

PyTorch

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

JAX (Core)

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

NumPy

API: —
Strategy: Custom / Partial

TensorFlow

API: tf.types.float8_e5m2
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

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