mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 19:06:07 +00:00
46e65b5982
I've left the TPU backend docs a stub for now. Hopefully, someone working on Pallas TPU can fill them in later.
16 lines
248 B
ReStructuredText
16 lines
248 B
ReStructuredText
``jax.experimental.pallas.tpu`` module
|
|
======================================
|
|
|
|
.. automodule:: jax.experimental.pallas.tpu
|
|
|
|
Classes
|
|
-------
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
Functions
|
|
---------
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary |