mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 12:46:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
943 Commits 508 Branches 263 Tags
Commit Graph

4 Commits

Author SHA1 Message Date
Peter Hawkins
b714cb30cc Add documentation for jax.jvp and jax.vjp. 2019-02-19 22:08:14 -05:00
Peter Hawkins
15e6c27130 Improve JAX API docs. 
Add examples for `jax.jit`, `jax.jacfwd`, and `jax.jacrev`.
Document `jax.hessian`. Add `argnums` support to `jax.hessian`.
 2019-02-15 08:16:25 -05:00
Matthew Johnson
9cf24029b2 add jax.random entry to jax.rst 2019-02-13 19:31:41 -08:00
Peter Hawkins
86d8915c3d Add Sphinx-generated reference documentation for JAX. 2019-01-16 09:13:31 -05:00