13 Commits

Author	SHA1	Message	Date
Matthew Johnson	bbf625e0df	fix jax.rst docs (remove defvjp2 / defjvp2)	2019-06-11 06:52:55 -07:00
Matthew Johnson	ab20f0292c	add docstring for defjvp_all	2019-06-05 17:34:14 -07:00
Matthew Johnson	720dec4072	add custom_gradient	2019-06-05 13:48:04 -07:00
Matthew Johnson	ffec059f0e	add jax.eval_shape to reference docs via jax.rst	2019-06-01 09:53:32 -07:00
Matthew Johnson	d199471696	add pmap to sphinx docs ... Co-authored-by: Peter Hawkins <phawkins@google.com>	2019-05-15 08:25:20 -07:00
Peter Hawkins	8c2c1a2e71	Expose jax.tree_util in the JAX docs.	2019-05-14 21:00:27 -04:00
Matthew Johnson	f5b4391f38	add jax.linearize to jax.readthedocs.io	2019-03-25 11:27:29 -07:00
Peter Hawkins	8b5e09f10a	Add new functions jax.ops.index_add and jax.ops.index_update for NumPy-style indexed updates. ... Create a new library `jax.ops` for user-facing ops that don't exist in NumPy or SciPy. Progress on issue #101. Fixes #122. Reenable some disabled TPU indexing tests that now pass.	2019-03-04 15:13:14 -05:00
Peter Hawkins	67174e3d57	Document jax.disable_jit. Add an example to jax.grad.	2019-02-20 09:00:12 -05:00
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