Extending jax MLIR with new dialect

[puzzleshark]

I have created a new MLIR dialect containing ops and types. I would like to be able to extend jax with my new dialect. Is there a way to do this without creating a fork of jax and including my dialect in the bazel build files?

I see that in the jaxlib/mlir folder there are the files libjaxlib_mlir_capi.so, _mlirDialectsSparseTensor.so, _mlirHlo.so, _mlir.so, _mlirSparseTensorPasses.so.

Is it possible for me to build a PYBIND entry point _myDialect.so and myDialect_capi.so and then link to the libjaxlib_mlir_capi.so?