https://github.com/google/jax
Tip revision: f4158ace933482844c145a6b919bf5dc86e084ba authored by jax authors on 18 June 2024, 12:21:34 UTC
Merge pull request #21949 from hawkinsp:winwheel
Merge pull request #21949 from hawkinsp:winwheel
Tip revision: f4158ac
File | Mode | Size |
---|---|---|
filecheck | ||
mosaic | ||
notebooks | ||
pallas | ||
testdata | ||
third_party | ||
BUILD | -rw-r--r-- | 30.8 KB |
ann_test.py | -rw-r--r-- | 7.1 KB |
aot_test.py | -rw-r--r-- | 3.7 KB |
api_test.py | -rw-r--r-- | 336.3 KB |
api_util_test.py | -rw-r--r-- | 2.4 KB |
array_api_test.py | -rw-r--r-- | 7.4 KB |
array_interoperability_test.py | -rw-r--r-- | 13.3 KB |
array_test.py | -rw-r--r-- | 52.5 KB |
attrs_test.py | -rw-r--r-- | 18.6 KB |
batching_test.py | -rw-r--r-- | 53.4 KB |
cache_key_test.py | -rw-r--r-- | 11.3 KB |
checkify_test.py | -rw-r--r-- | 40.6 KB |
cholesky_update_test.py | -rw-r--r-- | 2.6 KB |
clear_backends_test.py | -rw-r--r-- | 1.2 KB |
compilation_cache_test.py | -rw-r--r-- | 15.9 KB |
core_test.py | -rw-r--r-- | 23.9 KB |
custom_linear_solve_test.py | -rw-r--r-- | 15.9 KB |
custom_object_test.py | -rw-r--r-- | 10.0 KB |
custom_root_test.py | -rw-r--r-- | 8.2 KB |
debug_nans_test.py | -rw-r--r-- | 7.8 KB |
debugger_test.py | -rw-r--r-- | 11.0 KB |
debugging_primitives_test.py | -rw-r--r-- | 37.9 KB |
deprecation_test.py | -rw-r--r-- | 2.0 KB |
dtypes_test.py | -rw-r--r-- | 36.6 KB |
dynamic_api_test.py | -rw-r--r-- | 60.7 KB |
errors_test.py | -rw-r--r-- | 11.8 KB |
experimental_rnn_test.py | -rw-r--r-- | 6.5 KB |
export_back_compat_test.py | -rw-r--r-- | 33.3 KB |
export_harnesses_multi_platform_test.py | -rw-r--r-- | 6.7 KB |
export_test.py | -rw-r--r-- | 68.8 KB |
extend_test.py | -rw-r--r-- | 4.5 KB |
fft_test.py | -rw-r--r-- | 16.2 KB |
for_loop_test.py | -rw-r--r-- | 13.6 KB |
fused_attention_stablehlo_test.py | -rw-r--r-- | 16.3 KB |
generated_fun_test.py | -rw-r--r-- | 8.0 KB |
gfile_cache_test.py | -rw-r--r-- | 2.8 KB |
gpu_memory_flags_test.py | -rw-r--r-- | 1.7 KB |
heap_profiler_test.py | -rw-r--r-- | 1.2 KB |
host_callback_test.py | -rw-r--r-- | 108.7 KB |
host_callback_to_tf_test.py | -rw-r--r-- | 8.6 KB |
image_test.py | -rw-r--r-- | 14.5 KB |
infeed_test.py | -rw-r--r-- | 4.1 KB |
jax_jit_test.py | -rw-r--r-- | 8.0 KB |
jax_to_ir_test.py | -rw-r--r-- | 5.3 KB |
jaxpr_effects_test.py | -rw-r--r-- | 31.7 KB |
jaxpr_util_test.py | -rw-r--r-- | 3.7 KB |
jet_test.py | -rw-r--r-- | 16.5 KB |
key_reuse_test.py | -rw-r--r-- | 23.5 KB |
lax_autodiff_test.py | -rw-r--r-- | 47.1 KB |
lax_control_flow_test.py | -rw-r--r-- | 96.4 KB |
lax_metal_test.py | -rw-r--r-- | 225.7 KB |
lax_numpy_einsum_test.py | -rw-r--r-- | 12.1 KB |
lax_numpy_indexing_test.py | -rw-r--r-- | 63.3 KB |
lax_numpy_operators_test.py | -rw-r--r-- | 34.5 KB |
lax_numpy_reducers_test.py | -rw-r--r-- | 36.4 KB |
lax_numpy_test.py | -rw-r--r-- | 239.4 KB |
lax_numpy_ufuncs_test.py | -rw-r--r-- | 12.4 KB |
lax_numpy_vectorize_test.py | -rw-r--r-- | 8.1 KB |
lax_scipy_sparse_test.py | -rw-r--r-- | 15.9 KB |
lax_scipy_special_functions_test.py | -rw-r--r-- | 9.9 KB |
lax_scipy_spectral_dac_test.py | -rw-r--r-- | 2.0 KB |
lax_scipy_test.py | -rw-r--r-- | 22.4 KB |
lax_test.py | -rw-r--r-- | 160.2 KB |
lax_vmap_op_test.py | -rw-r--r-- | 2.9 KB |
lax_vmap_test.py | -rw-r--r-- | 29.5 KB |
layout_test.py | -rw-r--r-- | 12.8 KB |
lazy_loader_test.py | -rw-r--r-- | 1.3 KB |
linalg_test.py | -rw-r--r-- | 79.9 KB |
lobpcg_test.py | -rw-r--r-- | 14.6 KB |
logging_test.py | -rw-r--r-- | 4.7 KB |
lru_cache_test.py | -rw-r--r-- | 4.8 KB |
memories_test.py | -rw-r--r-- | 45.7 KB |
mesh_utils_test.py | -rw-r--r-- | 22.9 KB |
metadata_test.py | -rw-r--r-- | 4.1 KB |
mock_gpu_test.py | -rw-r--r-- | 1.6 KB |
monitoring_test.py | -rw-r--r-- | 6.2 KB |
mosaic_test.py | -rw-r--r-- | 1.1 KB |
multi_device_test.py | -rw-r--r-- | 12.1 KB |
multibackend_test.py | -rw-r--r-- | 7.3 KB |
multiprocess_gpu_test.py | -rw-r--r-- | 19.7 KB |
mutable_array_test.py | -rw-r--r-- | 6.4 KB |
name_stack_test.py | -rw-r--r-- | 21.3 KB |
nn_test.py | -rw-r--r-- | 17.2 KB |
ode_test.py | -rw-r--r-- | 8.5 KB |
optimizers_test.py | -rw-r--r-- | 10.0 KB |
package_structure_test.py | -rw-r--r-- | 1.9 KB |
pgle_test.py | -rw-r--r-- | 8.9 KB |
pickle_test.py | -rw-r--r-- | 6.1 KB |
pjit_test.py | -rw-r--r-- | 170.8 KB |
pmap_test.py | -rw-r--r-- | 118.0 KB |
polynomial_test.py | -rw-r--r-- | 4.3 KB |
pretty_printer_test.py | -rw-r--r-- | 1.2 KB |
profiler_test.py | -rw-r--r-- | 8.2 KB |
python_callback_test.py | -rw-r--r-- | 39.5 KB |
pytorch_interoperability_test.py | -rw-r--r-- | 5.6 KB |
qdwh_test.py | -rw-r--r-- | 7.3 KB |
random_lax_test.py | -rw-r--r-- | 56.3 KB |
random_test.py | -rw-r--r-- | 58.2 KB |
scipy_fft_test.py | -rw-r--r-- | 4.3 KB |
scipy_interpolate_test.py | -rw-r--r-- | 2.3 KB |
scipy_ndimage_test.py | -rw-r--r-- | 5.3 KB |
scipy_optimize_test.py | -rw-r--r-- | 6.8 KB |
scipy_signal_test.py | -rw-r--r-- | 16.2 KB |
scipy_spatial_test.py | -rw-r--r-- | 12.7 KB |
scipy_stats_test.py | -rw-r--r-- | 60.1 KB |
shape_poly_test.py | -rw-r--r-- | 148.8 KB |
shard_alike_test.py | -rw-r--r-- | 8.1 KB |
shard_map_test.py | -rw-r--r-- | 74.3 KB |
source_info_test.py | -rw-r--r-- | 1.7 KB |
sourcemap_test.py | -rw-r--r-- | 2.3 KB |
sparse_bcoo_bcsr_test.py | -rw-r--r-- | 76.6 KB |
sparse_nm_test.py | -rw-r--r-- | 7.0 KB |
sparse_test.py | -rw-r--r-- | 44.5 KB |
sparsify_test.py | -rw-r--r-- | 23.0 KB |
stack_test.py | -rw-r--r-- | 1.5 KB |
state_test.py | -rw-r--r-- | 66.0 KB |
stax_test.py | -rw-r--r-- | 8.0 KB |
svd_test.py | -rw-r--r-- | 10.8 KB |
transfer_guard_test.py | -rw-r--r-- | 8.5 KB |
tree_util_test.py | -rw-r--r-- | 42.6 KB |
typing_test.py | -rw-r--r-- | 5.8 KB |
util_test.py | -rw-r--r-- | 6.2 KB |
version_test.py | -rw-r--r-- | 6.6 KB |
x64_context_test.py | -rw-r--r-- | 4.4 KB |
xla_bridge_test.py | -rw-r--r-- | 14.3 KB |
xla_interpreter_test.py | -rw-r--r-- | 1.1 KB |
xmap_test.py | -rw-r--r-- | 77.3 KB |