mirror of
https://github.com/ROCm/jax.git
synced 2025-04-19 05:16:06 +00:00
15e6c27130
Add examples for `jax.jit`, `jax.jacfwd`, and `jax.jacrev`. Document `jax.hessian`. Add `argnums` support to `jax.hessian`.
23 lines
337 B
ReStructuredText
23 lines
337 B
ReStructuredText
jax package
|
|
===========
|
|
|
|
Subpackages
|
|
-----------
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
|
|
jax.numpy
|
|
jax.scipy
|
|
jax.experimental
|
|
jax.lax
|
|
jax.random
|
|
|
|
Module contents
|
|
---------------
|
|
|
|
.. automodule:: jax
|
|
:members: jit, grad, value_and_grad, vmap, jacfwd, jacrev, hessian, make_jaxpr
|
|
:undoc-members:
|
|
:show-inheritance:
|