mirror of
https://github.com/ROCm/jax.git
synced 2025-04-19 05:16:06 +00:00
1fa66590d1
The docstring for the recently added `pycapsule` function in `jax.extend.ffi` didn't conform to our usual docstring format, so I updated it and added a little bit more context.
18 lines
230 B
ReStructuredText
18 lines
230 B
ReStructuredText
.. currentmodule:: jax.extend
|
|
|
|
``jax.extend`` module
|
|
=====================
|
|
|
|
.. automodule:: jax.extend
|
|
|
|
Modules
|
|
-------
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
|
|
jax.extend.ffi
|
|
jax.extend.linear_util
|
|
jax.extend.mlir
|
|
jax.extend.random
|