rocm_jax

benchmarks

Cleanup: remove unused imports in tests (#3276 )

2020-06-01 11:49:35 -07:00

notebooks

Add Colab test notebooks for CPU, GPU, and TPU (#3000 )

2020-05-08 11:11:42 -07:00

third_party/scipy

BFGS algorithm (#3101 )

2020-07-29 14:14:40 -07:00

api_test.py

Revert "Revert "Delete batching.last. (#4148 )" (#4160 )" (#4242 )

2020-09-10 09:38:14 -04:00

api_util_test.py

Add ability to specify individual test targets

2020-06-29 11:08:57 -07:00

array_interoperability_test.py

Cleanup: define type lists in test_util & use in several test files. (#3616 )

2020-07-07 17:01:38 -07:00

batching_test.py

Add support for axis_index inside vmap (#4168 )

2020-08-28 20:03:39 +02:00

callback_test.py

Add ability to specify individual test targets

2020-06-29 11:08:57 -07:00

core_test.py

omnistaging, under a flag and disabled by default (#3370 )

2020-07-30 12:59:36 -07:00

debug_nans_test.py

Add ability to specify individual test targets

2020-06-29 11:08:57 -07:00

doubledouble_test.py

Remove some non-inclusive language (#3710 )

2020-07-10 09:29:06 -07:00

dtypes_test.py

Remove type restrictions (#4017 )

2020-08-10 20:48:20 -07:00

errors_test.py

unrevert #3674 (revert #3791 )

2020-08-17 18:13:58 -07:00

fft_test.py

hfft and ihfft implementation (#3664 )

2020-07-10 10:34:59 -07:00

generated_fun_test.py

Fix flaky generated_fun_test.py test (#3885 )

2020-07-28 07:27:07 -07:00

gmap_test.py

Add support for binding axis_name in gmap

2020-08-28 14:42:01 +02:00

host_callback_test.py

Cleanup: avoid jnp.prod & np.prod on array shapes (#4086 )

2020-08-18 10:17:38 -07:00

image_test.py

Implement nearest neighbor image resizes. (#3743 )

2020-07-13 20:27:12 -04:00

infeed_test.py

omnistaging, under a flag and disabled by default (#3370 )

2020-07-30 12:59:36 -07:00

jax_jit_test.py

Add more features to the C++ jax.jit. (#4169 )

2020-09-01 10:34:47 +03:00

jax_to_hlo_test.py

fix missing imports

2020-06-29 11:08:57 -07:00

jaxpr_util_test.py

fix jaxpr util test under enable_x64

2020-08-19 08:28:56 -07:00

jet_test.py

cumulative jet rules (#4000 )

2020-08-11 07:09:54 -07:00

lax_autodiff_test.py

Cleanup: avoid jnp.prod & np.prod on array shapes (#4086 )

2020-08-18 10:17:38 -07:00

lax_control_flow_test.py

Add reverse flag in associative scan (#4181 )

2020-09-04 09:21:43 -07:00

lax_numpy_einsum_test.py

Add ability to specify individual test targets

2020-06-29 11:08:57 -07:00

lax_numpy_indexing_test.py

Fixing weird behavior in segment_sum when num_segments is None (#4034 )

2020-09-11 13:51:42 -04:00

lax_numpy_test.py

implement jnp.apply_along_axis (#4253 )

2020-09-11 08:47:05 -07:00

lax_numpy_vectorize_test.py

Add ability to specify individual test targets

2020-06-29 11:08:57 -07:00

lax_scipy_sparse_test.py

Refine argument validation inside jax.scipy.sparse.linalg.cg (#3630 )

2020-07-06 09:24:44 -07:00

lax_scipy_test.py

Add Bessel functions in jax.numpy & jax.scipy.special (#4007 )

2020-08-10 10:10:59 -07:00

lax_test.py

improve lax.pad shape rule (#4234 )

2020-09-08 21:14:25 -07:00

lax_vmap_test.py

Add support for base dilation and window dilation to reduce window op… (#3803 )

2020-07-20 17:27:24 -04:00

linalg_test.py

Cleanup: make skip_if_unsupported_type more robust (#3912 )

2020-07-30 11:07:56 -07:00

loops_test.py

unrevert #3674 (revert #3791 )

2020-08-17 18:13:58 -07:00

masking_test.py

Addition of one more conclusive polynomial comparison case. (#4167 )

2020-08-28 17:27:32 +03:00

metadata_test.py

omnistaging, under a flag and disabled by default (#3370 )

2020-07-30 12:59:36 -07:00

multi_device_test.py

omnistaging, under a flag and disabled by default (#3370 )

2020-07-30 12:59:36 -07:00

multibackend_test.py

Add ability to specify individual test targets

2020-06-29 11:08:57 -07:00

nn_test.py

omnistaging, under a flag and disabled by default (#3370 )

2020-07-30 12:59:36 -07:00

ode_test.py

Fix scan carry types in gradient of complex ODE (#4130 )

2020-08-24 13:50:44 -07:00

optimizers_test.py

Add ability to specify individual test targets

2020-06-29 11:08:57 -07:00

optix_test.py

fix missing imports

2020-06-29 11:08:57 -07:00

pmap_test.py

Extend axis env while translating the pmapped jaxpr to XLA

2020-09-11 17:56:32 +02:00

polynomial_test.py

Fix polynomial tests (#3933 )

2020-08-01 08:37:31 -07:00

profiler_test.py

fix missing imports

2020-06-29 11:08:57 -07:00

random_test.py

allow random.choice to accept ndarray input (#4145 )

2020-08-26 10:21:56 -07:00

scipy_ndimage_test.py

Cleanup: avoid jnp.prod & np.prod on array shapes (#4086 )

2020-08-18 10:17:38 -07:00

scipy_optimize_test.py

Fixes to test_scipy_optimize.py for Google internal tests (#3902 )

2020-07-29 15:31:02 -07:00

scipy_signal_test.py

Cleanup: convert uses of 'import numpy as onp' in tests (#3756 )

2020-07-14 13:03:24 -07:00

scipy_stats_test.py

Cleanup: convert uses of 'import numpy as onp' in tests (#3756 )

2020-07-14 13:03:24 -07:00

sharded_jit_test.py

Avoid calling jnp.sum() on list (#4163 )

2020-08-28 09:07:30 -07:00

stax_test.py

Add ability to specify individual test targets

2020-06-29 11:08:57 -07:00

tree_util_tests.py

Compare treedefs by num_leaves not traversal_ in tree_transpose. (#3659 )

2020-07-30 13:31:17 -04:00

util_test.py

Add ability to specify individual test targets

2020-06-29 11:08:57 -07:00

vectorize_test.py

Add ability to specify individual test targets

2020-06-29 11:08:57 -07:00

xla_bridge_test.py

fix missing imports

2020-06-29 11:08:57 -07:00