jax
fa04e2d3 - #sdy #mixed_serialization don't make JAX export use `SdyRoundTripExportPipeline` to stringify attributes and convert ops to StableHLO `CustomCallOp`s and back.

Commit
224 days ago
#sdy #mixed_serialization don't make JAX export use `SdyRoundTripExportPipeline` to stringify attributes and convert ops to StableHLO `CustomCallOp`s and back. Instead keep every op as is now that StableHLO supports serialization with different dialects in the MLIR module. PiperOrigin-RevId: 778844006
Author
Parents
Loading