mirror of
https://github.com/ROCm/jax.git
synced 2025-04-25 09:56:08 +00:00
22 lines
313 B
ReStructuredText
22 lines
313 B
ReStructuredText
|
jax package
|
||
|
|
===========
|
||
|
|
|
||
|
|
Subpackages
|
||
|
|
-----------
|
||
|
|
|
||
|
|
.. toctree::
|
||
|
|
:maxdepth: 1
|
||
|
|
|
||
|
|
jax.numpy
|
||
|
|
jax.scipy
|
||
|
|
jax.experimental
|
||
|
|
jax.lax
|
||
|
|
|
||
|
|
Module contents
|
||
|
|
---------------
|
||
|
|
|
||
|
|
.. automodule:: jax
|
||
|
|
:members: jit, grad, value_and_grad, vmap, jacfwd, jacrev, make_jaxpr
|
||
|
|
:undoc-members:
|
||
|
|
:show-inheritance:
|