mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

add pmap to sphinx docs

Browse Source 
Co-authored-by: Peter Hawkins <phawkins@google.com>
This commit is contained in:
Matthew Johnson 2019-05-15 08:21:12 -07:00
parent 38c6a8e899
commit d199471696
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/jax.rst
View File

@ -19,6 +19,6 @@ Module contents
---------------
.. automodule:: jax
:members: jit, disable_jit, grad, value_and_grad, vmap, jacfwd, jacrev, hessian, jvp, linearize, vjp, make_jaxpr
:members: jit, disable_jit, grad, value_and_grad, vmap, pmap, jacfwd, jacrev, hessian, jvp, linearize, vjp, make_jaxpr
:undoc-members:
:show-inheritance: