Justin Lebar d5ba04b79e Add jax.ops.index_min/max. ...

These are analogous to index_add.

2019-06-21 19:33:34 -07:00

..

experimental

Fix reference to init_fun in optimizers docs.

2019-06-10 18:27:51 +01:00

interpreters

Return correct type from device_put.

2019-06-21 11:55:09 -04:00

lax

Add jax.ops.index_min/max.

2019-06-21 19:33:34 -07:00

lib

Remove xla.device_put_many.

2019-06-19 20:07:09 -04:00

numpy

Set copy=False in asarray.

2019-06-21 14:02:11 -04:00

ops

Add jax.ops.index_min/max.

2019-06-21 19:33:34 -07:00

scipy

fix build error

2019-06-11 07:15:14 -07:00

tools

Fix bazel warning about jax_to_hlo genrule.

2019-06-17 11:41:51 -07:00

__init__.py

Use a regular import to add jax.__version__ rather than exec() trickery.

2019-02-19 11:38:28 -05:00

abstract_arrays.py

improve literal hashing logic

2019-06-19 10:32:55 -07:00

ad_util.py

ship it

2019-05-08 13:41:25 -07:00

api_util.py

avoid packing leaf outputs for jit/pmap funs

2019-05-17 07:36:52 -07:00

api.py

make DeviceArray.__hash__ raise an error

2019-06-19 10:12:13 -07:00

BUILD

Add lax/*.py to jax library dependencies.

2019-04-16 09:29:13 -04:00

config.py

Test case improvements:

2019-04-12 10:48:11 -04:00

core.py

improve literal hashing logic

2019-06-19 10:32:55 -07:00

flatten_util.py

prevent jit from treating keyword args as static

2019-04-10 22:09:14 -07:00

lax_linalg.py

Fix complex Cholesky decomposition gradient:

2019-06-19 09:29:33 -04:00

lax_reference.py

source sync

2018-11-21 20:22:51 -08:00

linear_util.py

remove assert that fails on python3 map objects

2019-04-10 22:17:54 -07:00

pprint_util.py

source sync

2018-11-21 20:22:54 -08:00

random.py

make threefry_2x32 not do any op-by-op stuff

2019-06-11 14:56:21 -07:00

test_util.py

t # This is a combination of 2 commits.

2019-06-16 07:50:19 -07:00

tree_util.py

namedtuple subclass transparency (fixes #806)

2019-06-03 07:22:32 -07:00

util.py

make DeviceArray.__hash__ raise an error

2019-06-19 10:12:13 -07:00

version.py

bump version number for pypi

2019-06-18 09:20:48 -07:00