float8_e5m2 data type

Defines an 8-bit floating point representation with:

• 1 sign bit
• 5 exponent bits, with bias of 15
• 2 mantissa bits
• IEEE 754-compliant, with NaN and +/-inf.
• Subnormal numbers when biased exponent is 0.

Data type name

The data type is specified as "float8_e5m2".

Configuration

None.

Fill value representation

The fill value is specified in the same way as the core IEEE 754 floating point numbers: https://zarr-specs.readthedocs.io/en/latest/v3/core/v3.0.html#fill-value

The constant "NaN" corresponds to a representation of "0x7e".

Codec compatibility

bytes

Encoded as a 1-byte value 0bSEEEEEMM. The "endian" parameter has no effect.

See also

• Bit layout described by https://arxiv.org/abs/2209.05433
• Specified as the E5M2 format by OpenCompute MX.
• Python implementation available at https://pypi.org/project/ml-dtypes/
• Implemented in LLVM/MLIR as Float8E5M2.

Current maintainers

• Jeremy Maitin-Shepard (@jbms), Google