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.custom_partitioning.rst
``jax.experimental.custom_partitioning`` module
===============================================

.. automodule:: jax.experimental.custom_partitioning

API
---

.. autofunction:: custom_partitioning
back to top