jax

34cd5b0d - [Mosaic GPU] Remove sub-byte conversion restriction

[Mosaic GPU] Remove sub-byte conversion restriction XLA:GPU recently changed its endianness to little endian to better match LLVM and the rest of the CUDA ecosystem, so we can lift the earlier restrictions. PiperOrigin-RevId: 737934373