jax

f1067f9a - [Mosaic GPU] Add missing assertion in `test_serialize_uses_bytecode_format`.