Peter Hawkins ac8de3360d Add docstring and TODOs.

2019-04-12 17:11:20 -04:00

..

experimental

finish transposed convolution implementation and tests

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

interpreters

Add docstring and TODOs.

2019-04-12 17:11:20 -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

Adding np.isrealobj #70 (#594)

2019-04-11 10:16:55 -07:00

ops

DOC: Fix typo in ops.index_update

2019-03-30 17:18:53 +09:00

scipy

Merge pull request #548 from fehiepsi/pareto

2019-03-31 06:15:45 -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

Add np.longlong to the set of JAX array types.

2019-04-01 15:49:12 -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

add warning, fix typo in kwargs test and bug

2019-04-11 06:58:09 -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

Refactor handling of XLA backends.

2019-03-29 11:09:56 -04:00

lax_parallel.py

Factor out parallel functionality from lax.py into lax_parallel.py.

2019-04-10 13:02:04 -07:00

lax_reference.py

source sync

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

lax.py

remove the current lax.scan implementation (#599)

2019-04-11 21:26:06 -07: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

Merge branch 'master' into gamma

2019-04-12 07:15:41 -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