``jax.extend.pallas`` module
============================

.. automodule:: jax.extend.pallas

.. autosummary::
  :toctree: _autosummary

  GridMapping
  register_lowering_rule