mirror of
https://github.com/ROCm/jax.git
synced 2025-04-26 17:26:06 +00:00
25 lines
495 B
ReStructuredText
25 lines
495 B
ReStructuredText
jax package
|
|
===========
|
|
|
|
Subpackages
|
|
-----------
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
|
|
jax.numpy
|
|
jax.scipy
|
|
jax.experimental
|
|
jax.lax
|
|
jax.ops
|
|
jax.random
|
|
jax.tree_util
|
|
|
|
Module contents
|
|
---------------
|
|
|
|
.. automodule:: jax
|
|
:members: jit, disable_jit, grad, value_and_grad, vmap, pmap, jacfwd, jacrev, hessian, jvp, linearize, vjp, make_jaxpr, eval_shape, custom_transforms, defjvp, defjvp_all, defvjp, defvjp_all, custom_gradient
|
|
:undoc-members:
|
|
:show-inheritance:
|