multimesh.jax.autoshard

multimesh.jax.autoshard(autoshard=True)

Opens a context where shardings constraints become logical shardings.

Parameters:

autoshard (bool) – optional, whether the context should use logical autosharding.