I am seeking help with the review of
PR-16585 that introduces support for the float8_e4m3 type. This type has already been merged into LLVM, MLIR, StableHLO, and ml_dtypes.
This XLA PR is crucial as it paves the way for integrating float8_e4m3 support into JAX.
Thank you for your time and consideration.
Best regards,
Alex