jax.experimental.pallas.tpu.with\_memory\_space\_constraint
===========================================================

.. currentmodule:: jax.experimental.pallas.tpu

.. autofunction:: with_memory_space_constraint