https://github.com/google/jax
Raw File
Tip revision: 9e62994bce7c7fcbb2f6a50c9ef89526cd2c2be6 authored by jax authors on 09 May 2024, 19:32:51 UTC
Merge pull request #21135 from hawkinsp:release
Tip revision: 9e62994
jax.experimental.pjit.rst
``jax.experimental.pjit`` module
================================

.. automodule:: jax.experimental.pjit

API
---

.. autofunction:: pjit
back to top