|
..
|
|
filecheck
|
|
mosaic
|
|
notebooks
|
|
pallas
|
|
testdata
|
|
third_party
|
|
BUILD
|
|
ann_test.py
|
|
aot_test.py
|
|
api_test.py
|
|
api_util_test.py
|
|
array_api_skips.txt
|
|
array_api_test.py
|
|
array_extensibility_test.py
|
|
array_interoperability_test.py
|
|
array_test.py
|
|
attrs_test.py
|
|
batching_test.py
|
|
blocked_sampler_test.py
|
|
buffer_callback_test.py
|
|
cache_key_test.py
|
|
checkify_test.py
|
|
cholesky_update_test.py
|
|
clear_backends_test.py
|
|
colocated_python_test.py
|
|
compilation_cache_test.py
|
|
config_test.py
|
|
core_test.py
|
|
cudnn_fusion_test.py
|
|
custom_api_test.py
|
|
custom_linear_solve_test.py
|
|
custom_partitioning_sharding_rule_test.py
|
|
custom_root_test.py
|
|
debug_info_test.py
|
|
debug_nans_test.py
|
|
debugger_test.py
|
|
debugging_primitives_test.py
|
|
deprecation_test.py
|
|
device_test.py
|
|
distributed_initialize_test.py
|
|
distributed_test.py
|
|
dtypes_test.py
|
|
dynamic_api_test.py
|
|
error_check_test.py
|
|
errors_test.py
|
|
experimental_rnn_test.py
|
|
export_back_compat_test.py
|
|
export_harnesses_multi_platform_test.py
|
|
export_test.py
|
|
extend_test.py
|
|
ffi_test.py
|
|
fft_test.py
|
|
for_loop_test.py
|
|
fused_attention_stablehlo_test.py
|
|
garbage_collection_guard_test.py
|
|
generated_fun_test.py
|
|
gpu_memory_flags_test.py
|
|
heap_profiler_test.py
|
|
hijax_test.py
|
|
image_test.py
|
|
infeed_test.py
|
|
jax_jit_test.py
|
|
jax_numpy_error_test.py
|
|
jax_to_ir_test.py
|
|
jaxpr_effects_test.py
|
|
jaxpr_util_test.py
|
|
jet_test.py
|
|
key_reuse_test.py
|
|
lax_autodiff_test.py
|
|
lax_control_flow_test.py
|
|
lax_metal_test.py
|
|
lax_numpy_einsum_test.py
|
|
lax_numpy_indexing_test.py
|
|
lax_numpy_operators_test.py
|
|
lax_numpy_reducers_test.py
|
|
lax_numpy_test.py
|
|
lax_numpy_ufuncs_test.py
|
|
lax_numpy_vectorize_test.py
|
|
lax_scipy_sparse_test.py
|
|
lax_scipy_special_functions_test.py
|
|
lax_scipy_spectral_dac_test.py
|
|
lax_scipy_test.py
|
|
lax_test.py
|
|
lax_vmap_op_test.py
|
|
lax_vmap_test.py
|
|
layout_test.py
|
|
lazy_loader_test.py
|
|
linalg_sharding_test.py
|
|
linalg_test.py
|
|
lobpcg_test.py
|
|
logging_test.py
|
|
lru_cache_test.py
|
|
magma_linalg_test.py
|
|
memories_test.py
|
|
mesh_utils_test.py
|
|
metadata_test.py
|
|
mock_gpu_test.py
|
|
mock_gpu_topology_test.py
|
|
monitoring_test.py
|
|
mosaic_test.py
|
|
multi_device_test.py
|
|
multibackend_test.py
|
|
multiprocess_gpu_test.py
|
|
mutable_array_test.py
|
|
name_stack_test.py
|
|
nn_test.py
|
|
ode_test.py
|
|
optimizers_test.py
|
|
package_structure_test.py
|
|
pgle_test.py
|
|
pickle_test.py
|
|
pjit_test.py
|
|
pmap_test.py
|
|
polynomial_test.py
|
|
pretty_printer_test.py
|
|
profiler_test.py
|
|
python_callback_test.py
|
|
pytorch_interoperability_test.py
|
|
qdwh_test.py
|
|
ragged_collective_test.py
|
|
random_lax_test.py
|
|
random_test.py
|
|
roofline_test.py
|
|
scaled_matmul_stablehlo_test.py
|
|
scipy_fft_test.py
|
|
scipy_interpolate_test.py
|
|
scipy_ndimage_test.py
|
|
scipy_optimize_test.py
|
|
scipy_signal_test.py
|
|
scipy_spatial_test.py
|
|
scipy_stats_test.py
|
|
shape_poly_test.py
|
|
shard_alike_test.py
|
|
shard_map_test.py
|
|
source_info_test.py
|
|
source_mapper_test.py
|
|
sourcemap_test.py
|
|
sparse_bcoo_bcsr_test.py
|
|
sparse_test.py
|
|
sparsify_test.py
|
|
stack_test.py
|
|
state_test.py
|
|
stax_test.py
|
|
string_array_test.py
|
|
svd_test.py
|
|
transfer_guard_test.py
|
|
tree_util_test.py
|
|
typing_test.py
|
|
unary_ops_accuracy_test.py
|
|
util_test.py
|
|
version_test.py
|
|
warnings_util_test.py
|
|
x64_context_test.py
|
|
xla_bridge_test.py
|
|
xla_interpreter_test.py
|
|
xla_metadata_test.py
|