Peter Hawkins
637a802e71
Improve layout of jax.scipy documentation.
...
Fixes #577 , since the set of supported distributions is now relatively easy to see.
2019-04-15 12:40:18 -04:00
Peter Hawkins
407306293f
Update lax documentation to reflect new code organization.
2019-04-15 12:16:14 -04:00
Aditya Vaidya
48934dc9d1
Implement cross product and test cases
2019-03-25 18:22:32 -05:00
Matthew Johnson
f5b4391f38
add jax.linearize to jax.readthedocs.io
2019-03-25 11:27:29 -07:00
Peter Hawkins
eefc0a7378
Add jax.ops to documentation.
2019-03-04 20:12:22 -05: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
fb8df7cc19
Add docstrings for lax.gather/scatter.
...
Also add `float_power` to the numpy docs.
2019-02-22 08:40:27 -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
7fc4e0237b
Implement np.gcd and np.lcm.
...
Taking the loop primitives out for a spin!
2019-02-19 15:57:22 -05:00
Peter Hawkins
febadd7354
Add np.roll to the documentation.
2019-02-19 07:29:30 -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
Matthew Johnson
c467bbfe50
add jax.random to docs
2019-02-13 19:21:55 -08:00
Peter Hawkins
175c779b93
Add np.linalg.norm to the docs.
2019-02-07 11:02:56 -05:00
Peter Hawkins
ff1f5fc647
Rename jax.experimental.minmax to jax.experimental.optimizers in the documentation.
2019-02-07 08:09:12 -05:00
Matthew Johnson
5847d96168
rename "minmax" -> "optimizers"
2019-02-06 11:02:16 -08:00
Peter Hawkins
9338d3d704
Implement np.heaviside.
2019-02-06 09:05:53 -05:00
Peter Hawkins
7edd1337f2
Add axis argument to np.stack. Implement np.{dstack,atleast_3d}.
2019-02-06 08:46:11 -05:00
Peter Hawkins
c5433bd892
Implement np.{empty,empty_like,ptp,isreal,iscomplex,sinc,vander,positive}.
...
Fix bug in definition of `np.imag` for real numbers.
Fix wrong output (pi vs 0) for `np.angle` for negative real numbers. Fix semantics of angle for integers.
Issue #70
2019-02-05 10:29:31 -05:00
Peter Hawkins
980ea88bfe
Forward np.{bartlett,blackman,hamming,hanning,kaiser} to numpy.
2019-02-04 21:26:58 -05:00
Peter Hawkins
91704ad287
Add new numpy functions to docs.
2019-02-04 10:45:50 -05:00
Peter Hawkins
fec5f47596
Add np.take to the documentation index.
2019-02-02 09:47:55 -05:00
Peter Hawkins
8d52ee899c
Add newly added Numpy functions to docs.
2019-02-01 08:57:22 -05:00
Peter Hawkins
326c3dbb99
Add license header to docs/conf.py.
2019-01-24 08:39:50 -05:00
Peter Hawkins
7d78c3c84c
Drop jaxlib dependency from setup.py, add a requirements.txt for readthedoc build instead.
2019-01-16 09:45:25 -05:00
Peter Hawkins
86d8915c3d
Add Sphinx-generated reference documentation for JAX.
2019-01-16 09:13:31 -05:00
