Float8_e5m2

Float8_e5m2 data type constructor.

Abstract Signature:

Float8_e5m2(x)

PyTorch

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

JAX (Core)

API: jax.numpy.float8_e5m2
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.numpy.float8_e5m2
Strategy: Direct Mapping
Official Docs ↗