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

..

conf.py

Add license header to docs/conf.py.

2019-01-24 08:39:50 -05:00

index.rst

Add Sphinx-generated reference documentation for JAX.

2019-01-16 09:13:31 -05:00

jax.experimental.optimizers.rst

Rename jax.experimental.minmax to jax.experimental.optimizers in the documentation.

2019-02-07 08:09:12 -05:00

jax.experimental.rst

rename "minmax" -> "optimizers"

2019-02-06 11:02:16 -08:00

jax.experimental.stax.rst

Add Sphinx-generated reference documentation for JAX.

2019-01-16 09:13:31 -05:00

jax.lax.rst

Add Sphinx-generated reference documentation for JAX.

2019-01-16 09:13:31 -05:00

jax.numpy.rst

Add docstrings for lax.gather/scatter.

2019-02-22 08:40:27 -05:00

jax.random.rst

add jax.random to docs

2019-02-13 19:21:55 -08:00

jax.rst

Add new functions jax.ops.index_add and jax.ops.index_update for NumPy-style indexed updates.

2019-03-04 15:13:14 -05:00

jax.scipy.rst

Add Sphinx-generated reference documentation for JAX.

2019-01-16 09:13:31 -05:00

modules.rst

Add Sphinx-generated reference documentation for JAX.

2019-01-16 09:13:31 -05:00

requirements.txt

Drop jaxlib dependency from setup.py, add a requirements.txt for readthedoc build instead.

2019-01-16 09:45:25 -05:00