James Bradbury a26963fe87

Merge pull request #1106 from google/jb/bool-reduction ...

fix jax.numpy reduction init_val for bools

2019-08-05 10:45:17 -07:00

..

api_test.py

Merge branch 'master' into jit-device-placement-api

2019-07-24 20:35:43 +01:00

batching_test.py

add batching rule for lax.sort

2019-08-01 12:39:33 -04:00

core_test.py

Fix test failures.

2019-08-01 17:20:27 -04:00

fft_test.py

Initial FFT support.

2019-05-16 14:37:30 -07:00

generated_fun_test.py

fix miscellaneous bugs, incl. complex abs grad

2018-12-12 12:03:57 -05:00

jax_to_hlo_test.py

Replace jaxlib import paths with jax.lib paths.

2019-07-30 21:01:48 -04:00

lapax_test.py

Update lapax_test.py

2019-02-13 21:56:52 +05:30

lax_control_flow_test.py

enable jit+pmap by merging pxla.py and xla.py

2019-07-05 16:39:46 -07:00

lax_numpy_einsum_test.py

tweak an einsum test

2019-04-08 09:52:47 -07:00

lax_numpy_indexing_test.py

Remove assertions in scatter/dynamic_update_slice JVP rules that test whether index tangents are symbolically zero.

2019-07-23 14:18:07 -04:00

lax_numpy_test.py

Fix test breakage at head.

2019-08-02 08:55:22 -04:00

lax_scipy_test.py

Disable gradient test for entr.

2019-08-02 14:17:31 -04:00

lax_test.py

Merge pull request #1106 from google/jb/bool-reduction

2019-08-05 10:45:17 -07:00

linalg_test.py

Remove commented lines from test.

2019-08-02 11:48:12 -04:00

optimizers_test.py

Fix pack_optimizer_state to correctly use tuples everywhere in the packed state and add a unit test to check round trip unpack/pack.

2019-08-02 16:42:17 -07:00

parallel_test.py

Fix flake8 warnings, including a few real bugs.

2019-08-04 10:25:48 -04:00

pmap_test.py

fix num_replicas counts for initial-style control

2019-07-25 18:17:57 -07:00

random_test.py

Fix flake8 warnings, including a few real bugs.

2019-08-04 10:25:48 -04:00

scipy_stats_test.py

Enable some tests that now pass.

2019-07-21 16:28:15 -04:00

stax_test.py

expose sigmoid too

2019-08-01 17:11:31 -04:00

tree_util_tests.py

Address review comments.

2019-08-01 16:48:18 -04:00