Skye Wanderman-Milne
5d1c014509
Initial FFT support.
...
This change creates a new fft primitive in lax, and uses it to implement numpy's np.fft.fftn function.
Not-yet-implemented functionality:
- vmap
- 's' argument of fftn
- other numpy np.fft functions
Resolves #505 .
2019-05-16 14:37:30 -07:00
Matthew Johnson
0e6aa3c4e3
Merge pull request #718 from google/pmap-docs
...
Add pmap docstring
2019-05-15 08:27:02 -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
8f461d265f
Add some missing special functions to the documentation.
2019-05-15 09:54:24 -04:00
Peter Hawkins
8c2c1a2e71
Expose jax.tree_util in the JAX docs.
2019-05-14 21:00:27 -04:00
Peter Hawkins
7e3295eb50
Add np.linalg.eig to the documentation.
2019-05-13 16:02:02 -04:00
Peter Hawkins
c2369ce340
Add lax.scan to docs.
2019-05-13 14:46:35 -04:00
Peter Hawkins
e6082d203d
Add segment_sum to the docs and fix its rendering.
...
Minor doc fixes.
2019-05-08 21:18:17 -04:00
Matthew Johnson
a0fcb3fb9d
Merge pull request #660 from Bharat123rox/npfix
...
Implemented np.fix
2019-04-30 16:56:03 -07:00
Sharad Vikram
ea63489516
added tile to docs
2019-04-30 13:31:34 -07:00
Bharat123rox
69d12111fc
Implemented np.fix
2019-04-30 22:33:25 +05:30
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
