Matthew Johnson 1262ca9b30 improve conv rhs batching, add systematic test

2019-06-15 12:01:20 -07:00

..

experimental

Fix reference to init_fun in optimizers docs.

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

interpreters

fix the david sussillo bug

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

lax

improve conv rhs batching, add systematic test

2019-06-15 12:01:20 -07:00

lib

revise nested pmap gather to CRS-based impl

2019-05-17 10:25:26 -07:00

numpy

make threefry_2x32 not do any op-by-op stuff

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

ops

Add segment_sum to the docs and fix its rendering.

2019-05-08 21:18:17 -04:00

scipy

fix build error

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

tools

Add jax_to_hlo.

2019-06-14 11:43: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

allow more types to be jaxpr literals, fixes #772

2019-05-28 22:38:06 -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

implement reviewer suggestions

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

undo #503 in favor of new literal staging method

2019-05-29 08:12:05 -07:00

flatten_util.py

prevent jit from treating keyword args as static

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

lax_linalg.py

Check for failure of LAPACK calls and return arrays full of NaNs if they occur.

2019-05-29 16:50:27 -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

improve conv rhs batching, add systematic test

2019-06-15 12:01:20 -07:00

tree_util.py

namedtuple subclass transparency (fixes #806)

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

util.py

make static_argnums cache on value when possible

2019-05-09 20:00:24 -07:00

version.py

bump version for pypi

2019-06-11 14:03:37 -07:00