| File | Mode | Size |
|---|---|---|
| api_test.py | -rw-r--r-- | 33.8 KB |
| batching_test.py | -rw-r--r-- | 38.1 KB |
| core_test.py | -rw-r--r-- | 6.5 KB |
| debug_nans_test.py | -rw-r--r-- | 1.6 KB |
| fft_test.py | -rw-r--r-- | 3.3 KB |
| generated_fun_test.py | -rw-r--r-- | 7.8 KB |
| jax_to_hlo_test.py | -rw-r--r-- | 2.4 KB |
| lapax_test.py | -rw-r--r-- | 6.9 KB |
| lax_control_flow_test.py | -rw-r--r-- | 32.3 KB |
| lax_numpy_einsum_test.py | -rw-r--r-- | 7.7 KB |
| lax_numpy_indexing_test.py | -rw-r--r-- | 37.0 KB |
| lax_numpy_test.py | -rw-r--r-- | 88.5 KB |
| lax_scipy_test.py | -rw-r--r-- | 5.9 KB |
| lax_test.py | -rw-r--r-- | 125.6 KB |
| linalg_test.py | -rw-r--r-- | 31.7 KB |
| masking_test.py | -rw-r--r-- | 10.3 KB |
| multibackend_test.py | -rw-r--r-- | 4.3 KB |
| nn_test.py | -rw-r--r-- | 2.8 KB |
| optimizers_test.py | -rw-r--r-- | 9.8 KB |
| parallel_test.py | -rw-r--r-- | 9.5 KB |
| pmap_test.py | -rw-r--r-- | 27.6 KB |
| random_test.py | -rw-r--r-- | 15.1 KB |
| scipy_stats_test.py | -rw-r--r-- | 10.0 KB |
| stax_test.py | -rw-r--r-- | 10.1 KB |
| tree_util_tests.py | -rw-r--r-- | 4.1 KB |
| vectorize_test.py | -rw-r--r-- | 5.2 KB |
