James Bradbury 23f06b417f add initializer tests

2019-10-03 12:01:21 -07:00

..

api_test.py

Add optional tuple_args argument to xla_computation.

2019-09-30 11:10:32 -07:00

batching_test.py

remove broadcasts from _dot_general_batch_rule

2019-09-10 13:58:23 -07:00

core_test.py

De-tuplify the rest of the core

2019-08-21 13:21:20 -07:00

debug_nans_test.py

test that debug_nans actually finds NaNs

2019-10-02 16:10:27 -07: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

Merge pull request #1339 from shoyer/solvers

2019-09-27 12:17:47 -07:00

lax_numpy_einsum_test.py

tweak an einsum test

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

lax_numpy_indexing_test.py

skip problematic tests on tpu

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

lax_numpy_test.py

nanmean + nanprod only work on floating-point types.

2019-09-27 11:27:24 -07:00

lax_scipy_test.py

Disable gradient test for entr.

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

lax_test.py

add some 'manual' lax.rem autodiff tests

2019-09-15 09:25:54 -07:00

linalg_test.py

Disable eigh test on TPU.

2019-09-24 18:23:09 -07:00

masking_test.py

tweaks to simplify masked jaxprs, rnn test

2019-09-16 15:47:43 -07:00

multibackend_test.py

clean up multibackend tests

2019-08-23 23:42:08 -07:00

nn_test.py

add initializer tests

2019-10-03 12:01:21 -07:00

optimizers_test.py

Add polynomial_decay schedule to optimizers

2019-08-30 18:33:58 +01:00

parallel_test.py

De-tuplify the rest of the core

2019-08-21 13:21:20 -07:00

pmap_test.py

Fix pmap error message.

2019-09-27 11:50:21 -07:00

random_test.py

address reviewer comments, no op-by-op in threefry

2019-08-13 11:30:24 -07: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

restore the behavior that Nones are pytrees

2019-08-23 15:47:42 -07:00

vectorize_test.py

fixing issues with tests

2019-08-30 20:36:54 +00:00