multimesh.jax.ignore_transforms

multimesh.jax.ignore_transforms(ignore=True)

Sets debug context where all MultiMesh transformations are no-ops.

To aid in debugging, functions can be invoked in a context in which all transformations like task and with_sharding_constraint are ignored. This essentially falls back to default Jax execution.

Parameters:

ignore (bool | None) – optional, whether MultiMesh transformations should be ignored