Float8E5m2 ========== FP8 type E5M2. .. raw:: html

PyTorch

API: torch.float8_e5m2
Strategy: Direct Mapping

JAX (Core)

API: jnp.float8_e5m2
Strategy: Direct Mapping

NumPy

API:
Strategy: Custom / Partial

TensorFlow

API: tf.types.float8_e5m2
Strategy: Direct Mapping

Flax NNX

API: jnp.float8_e5m2
Strategy: Direct Mapping