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 jax.linearize to jax.readthedocs.io

Browse Source
This commit is contained in:
Matthew Johnson 2019-03-25 11:27:29 -07:00
parent a9ab470ec5
commit f5b4391f38
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

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