mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 19:06:07 +00:00
10 lines
138 B
ReStructuredText
10 lines
138 B
ReStructuredText
``jax.experimental.pjit`` module
|
|
================================
|
|
|
|
.. automodule:: jax.experimental.pjit
|
|
|
|
API
|
|
---
|
|
|
|
.. autofunction:: pjit
|