jax.experimental.pjit module
============================

.. automodule:: jax.experimental.pjit

API
---

.. autofunction:: pjit