Matthew Johnson 08dc6994f5 partial progress

2019-02-01 17:05:49 -08:00

..

experimental

Fix implementation of average pooling to align the window element counts with the spatial dimensions.

2019-01-28 09:20:02 -05:00

interpreters

partial progress

2019-02-01 17:05:49 -08:00

lib

Make translation rule for select_and_gather_add work even when --jax_enable_x64 is disabled.

2019-01-28 18:41:27 -05:00

numpy

Fix gradient for np.amin and np.amax.

2019-02-01 11:53:12 -05:00

scipy

Merge branch 'master' of https://github.com/google/jax into svd

2019-01-08 09:30:05 +05:30

__init__.py

Set default TF log level to "1" to avoid reporting things like CPU frequency at import time. Also import jax.numpy in __init__.py because it has side effects that set up the infix operator overloading.

2018-12-05 15:55:01 -05:00

abstract_arrays.py

Add preliminary support for np.complex128.

2019-01-11 18:22:43 -05:00

ad_util.py

source sync

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

api_util.py

tweak

2019-01-28 09:19:06 -08:00

api.py

partial progress

2019-02-01 17:05:49 -08:00

BUILD

source sync

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

config.py

fix failing tests (misc small bugs)

2018-12-13 11:52:41 -08:00

core.py

make tuple unpacking cause a full_lower

2019-01-07 16:47:13 -08:00

flatten_util.py

generalize jacfwd and jacrev to handle pytrees

2019-01-06 12:49:41 -08:00

lax_linalg.py

Add complex128 support to LAPACK bindings.

2019-01-18 16:54:07 -05:00

lax_reference.py

source sync

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

lax.py

partial progress

2019-02-01 17:05:49 -08:00

linear_util.py

Increase cache size to 4096.

2019-01-15 10:32:58 -05:00

pprint_util.py

source sync

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

random.py

set behavior when random.randint has invalid range

2019-01-12 13:12:40 -08:00

test_util.py

Fix complex64/complex128 confusion in test harness.

2019-02-01 14:01:06 -05:00

tree_util.py

partial progress

2019-02-01 17:05:49 -08:00

util.py

Increase cache size to 4096.

2019-01-15 10:32:58 -05:00