| .. |
|
__main__.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
cuda_skip.py
|
Align mlx::core::max op nan propagation with NumPy (#2339)
|
2025-07-09 11:26:27 -07:00 |
|
mlx_distributed_tests.py
|
Distributed layers (#1270)
|
2025-03-21 13:52:17 -07:00 |
|
mlx_tests.py
|
use fp32 for testing, add more complex ops (#2322)
|
2025-07-01 07:30:00 -07:00 |
|
mpi_test_distributed.py
|
Fix the test and add custom min/max reductions for uncommon MPI types (#2060)
|
2025-04-10 17:01:17 -07:00 |
|
ring_test_distributed.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_array.py
|
[CUDA] Fix back-end bugs and enable corresponding tests (#2296)
|
2025-06-16 08:45:40 -07:00 |
|
test_autograd.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_bf16.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_blas.py
|
MoE backward improvements (#2335)
|
2025-07-07 17:59:53 -07:00 |
|
test_compile.py
|
Split broadcast so it is always fused in compile (#2318)
|
2025-06-26 22:08:18 -07:00 |
|
test_constants.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_conv_transpose.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_conv.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_device.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_double.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_einsum.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_eval.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_export_import.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_fast_sdpa.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_fast.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_fft.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_graph.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_init.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_linalg.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_load.py
|
[CUDA] Switch to CUDA graphs (#2317)
|
2025-07-02 15:59:13 -07:00 |
|
test_losses.py
|
Cuda bug fixes 2 (#2298)
|
2025-06-16 13:14:46 -07:00 |
|
test_memory.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_nn.py
|
allow parameters to be deleted (#2325)
|
2025-07-01 21:27:23 -07:00 |
|
test_ops.py
|
[CUDA] Fix back-end bugs and enable corresponding tests (#2296)
|
2025-06-16 08:45:40 -07:00 |
|
test_optimizers.py
|
Fix type promotion in Adam with bias correction (#2350)
|
2025-07-10 11:14:42 -07:00 |
|
test_quantized.py
|
MoE backward improvements (#2335)
|
2025-07-07 17:59:53 -07:00 |
|
test_random.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_reduce.py
|
Align mlx::core::min op nan propagation with NumPy (#2346)
|
2025-07-10 06:20:43 -07:00 |
|
test_tree.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_upsample.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
|
test_vmap.py
|
add python testing for cuda with ability to skip list of tests (#2295)
|
2025-06-15 10:56:48 -07:00 |
