Skip to main content

Browse the archive

https://github.com/google/jax
17 August 2024, 15:35:15 UTC
History
Tip revision: 3e5466c2f9c21edfad8a7ea1e973aec1a763d64c authored by jax authors on 18 November 2021, 17:19:17 UTC
Internal change.
Tip revision: 3e5466c
File Mode Size
benchmarks
notebooks
third_party
ann_test.py -rw-r--r-- 4.1 KB
api_test.py -rw-r--r-- 196.5 KB
api_util_test.py -rw-r--r-- 2.4 KB
array_interoperability_test.py -rw-r--r-- 8.2 KB
batching_test.py -rw-r--r-- 51.4 KB
callback_test.py -rw-r--r-- 7.1 KB
compilation_cache_test.py -rw-r--r-- 14.1 KB
core_test.py -rw-r--r-- 14.7 KB
custom_object_test.py -rw-r--r-- 11.4 KB
debug_nans_test.py -rw-r--r-- 6.6 KB
djax_test.py -rw-r--r-- 4.8 KB
dtypes_test.py -rw-r--r-- 17.1 KB
errors_test.py -rw-r--r-- 10.9 KB
fft_test.py -rw-r--r-- 15.8 KB
file_system_cache_test.py -rw-r--r-- 5.5 KB
generated_fun_test.py -rw-r--r-- 8.0 KB
gsda_test.py -rw-r--r-- 10.8 KB
heap_profiler_test.py -rw-r--r-- 1.2 KB
host_callback_test.py -rw-r--r-- 101.2 KB
host_callback_to_tf_test.py -rw-r--r-- 7.7 KB
image_test.py -rw-r--r-- 16.6 KB
infeed_test.py -rw-r--r-- 3.7 KB
jax_jit_test.py -rw-r--r-- 7.7 KB
jax_to_hlo_test.py -rw-r--r-- 2.4 KB
jaxpr_util_test.py -rw-r--r-- 2.7 KB
jet_test.py -rw-r--r-- 15.3 KB
lax_autodiff_test.py -rw-r--r-- 50.9 KB
lax_control_flow_test.py -rw-r--r-- 92.9 KB
lax_numpy_einsum_test.py -rw-r--r-- 8.9 KB
lax_numpy_indexing_test.py -rw-r--r-- 51.1 KB
lax_numpy_test.py -rw-r--r-- 269.7 KB
lax_numpy_vectorize_test.py -rw-r--r-- 7.5 KB
lax_scipy_sparse_test.py -rw-r--r-- 16.8 KB
lax_scipy_test.py -rw-r--r-- 24.5 KB
lax_test.py -rw-r--r-- 123.3 KB
lax_vmap_test.py -rw-r--r-- 35.2 KB
linalg_test.py -rw-r--r-- 59.6 KB
loops_test.py -rw-r--r-- 14.7 KB
masking_test.py -rw-r--r-- 30.6 KB
metadata_test.py -rw-r--r-- 4.1 KB
multi_device_test.py -rw-r--r-- 8.0 KB
multibackend_test.py -rw-r--r-- 7.1 KB
nn_test.py -rw-r--r-- 9.1 KB
ode_test.py -rw-r--r-- 7.9 KB
optimizers_test.py -rw-r--r-- 10.1 KB
pickle_test.py -rw-r--r-- 2.0 KB
pjit_test.py -rw-r--r-- 38.5 KB
pmap_test.py -rw-r--r-- 92.8 KB
polynomial_test.py -rw-r--r-- 5.4 KB
profiler_test.py -rw-r--r-- 6.1 KB
qdwh_test.py -rw-r--r-- 6.7 KB
random_test.py -rw-r--r-- 51.7 KB
scipy_fft_test.py -rw-r--r-- 3.3 KB
scipy_ndimage_test.py -rw-r--r-- 6.0 KB
scipy_optimize_test.py -rw-r--r-- 6.9 KB
scipy_signal_test.py -rw-r--r-- 4.6 KB
scipy_stats_test.py -rw-r--r-- 22.4 KB
sharded_jit_test.py -rw-r--r-- 17.0 KB
sparse_test.py -rw-r--r-- 67.5 KB
sparsify_test.py -rw-r--r-- 11.5 KB
stax_test.py -rw-r--r-- 10.8 KB
tree_util_test.py -rw-r--r-- 13.5 KB
util_test.py -rw-r--r-- 2.5 KB
x64_context_test.py -rw-r--r-- 5.1 KB
xla_bridge_test.py -rw-r--r-- 8.0 KB
xla_interpreter_test.py -rw-r--r-- 1.1 KB
xmap_test.py -rw-r--r-- 56.4 KB
back to top