Float8E4m3fn

FP8 type E4M3FN.

PyTorch

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

JAX (Core)

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

NumPy

API: —
Strategy: Custom / Partial

TensorFlow

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

Flax NNX

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