PyTorch
API: torch.float8_e4m3fn
Strategy: Direct Mapping
JAX (Core)
API: jnp.float8_e4m3fn
Strategy: Direct Mapping
NumPy
API: —
Strategy: Custom / Partial
TensorFlow
API: tf.types.float8_e4m3fn
Strategy: Direct Mapping
Flax NNX
API: jnp.float8_e4m3fn
Strategy: Direct Mapping