jax.lax.with\_sharding\_constraint
==================================

.. currentmodule:: jax.lax

.. autofunction:: with_sharding_constraint