How to disable jit when using scalax wrapper

[Vandermode]

Hi~ Thank you for your amazing work!

I have a question about debugging when using this lib.
Previously, I used jax.config.update("jax_disable_jit", True) to disable jit functionality for debugging.
However, this instruction seems no longer effective when my function is wrapped by @partial(mesh.sjit) .
Could you please provide me some guidance on how to globally disable the jit without changing the code?

Thank you!