jax

c7a3bb5a - [Mosaic GPU] Fix a MLIR error when doing layout conversions involving f8 types

[Mosaic GPU] Fix a MLIR error when doing layout conversions involving f8 types They are not representable in LLVM IR and special care has to be taken to ensure that the unrealized conversion casts cancel out. PiperOrigin-RevId: 862706717