Peter Hawkins 407306293f Update lax documentation to reflect new code organization.

2019-04-15 12:16:14 -04:00

..

experimental

finish transposed convolution implementation and tests

2019-04-09 22:59:03 -07:00

interpreters

misc python performance improvements

2019-04-15 07:45:10 -07:00

lax

Update lax documentation to reflect new code organization.

2019-04-15 12:16:14 -04:00

lib

Add a device_put_many() method that copies multiple tensors to accelerator devices in parallel. Use it to copy pmap arguments.

2019-04-12 16:18:48 -04:00

numpy

Use lax.sign instead of lax.div(x, lax.abs(x))

2019-04-14 22:42:52 +05:30

ops

make a lax package, revert control flow names (#607)

2019-04-12 16:28:40 -07:00

scipy

make a lax package, revert control flow names (#607)

2019-04-12 16:28:40 -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

misc python performance improvements

2019-04-15 07:45:10 -07:00

ad_util.py

peval.py works again (some paired w/ @dougalm)

2019-02-22 07:53:28 -08:00

api_util.py

prevent jit from treating keyword args as static

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

api.py

make jacrev work w/ complex inputs, update errors (#610)

2019-04-13 13:22:45 -07:00

BUILD

Add missing jax/ops/ to jax/BUILD.

2019-03-06 09:07:15 -05:00

config.py

Test case improvements:

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

core.py

prevent jit from treating keyword args as static

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

flatten_util.py

prevent jit from treating keyword args as static

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

lax_linalg.py

make a lax package, revert control flow names (#607)

2019-04-12 16:28:40 -07: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 a lax package, revert control flow names (#607)

2019-04-12 16:28:40 -07:00

test_util.py

Test case improvements:

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

tree_util.py

comment-out new pytree check with a TODO

2019-03-22 07:38:40 -07:00

util.py

fix pmap performance bug, dont always copy to host

2019-03-26 08:35:34 -07:00

version.py

bump version for pypi

2019-04-07 17:50:43 -07:00