jax.lax.reduce_precision

jax.lax.reduce_precision(operand, exponent_bits, mantissa_bits)[source]

Wraps XLA’s ReducePrecision operator.