Luca Vivona 728d4db582 Support destination arg in tree flatten/unflatten (#2450)		2025-08-06 15:34:59 -07:00
..
__main__.py
cuda_skip.py	[CUDA] Backward convolution (#2431)	2025-08-01 09:54:05 +09:00
mlx_distributed_tests.py
mlx_tests.py
mpi_test_distributed.py
ring_test_distributed.py	fix ring distributed test (#2380)	2025-07-16 11:25:24 -07:00
test_array.py
test_autograd.py	Fix imag() vjp (#2367)	2025-07-14 13:11:16 -07:00
test_bf16.py
test_blas.py	fix gemv regression (#2445)	2025-07-30 14:23:01 -07:00
test_compile.py	fix wraps compile (#2461)	2025-08-04 16:14:18 -07:00
test_constants.py
test_conv_transpose.py
test_conv.py
test_device.py
test_double.py
test_einsum.py
test_eval.py
test_export_import.py	Add tests for export including control flow models and quantized models (#2430)	2025-07-31 11:06:26 -07:00
test_fast_sdpa.py	full row mask in sdpa consistently gives nan (#2406)	2025-07-23 16:37:03 -07:00
test_fast.py
test_fft.py
test_graph.py
test_init.py
test_linalg.py
test_load.py	[CUDA] Switch to CUDA graphs (#2317)	2025-07-02 15:59:13 -07:00
test_losses.py
test_memory.py
test_nn.py	Support destination arg in tree flatten/unflatten (#2450)	2025-08-06 15:34:59 -07:00
test_ops.py	faster rms norm (#2433)	2025-07-29 13:12:00 -07:00
test_optimizers.py	Adding support for the Muon Optimizer (#1914)	2025-07-18 12:25:28 -07:00
test_quantized.py	faster gather qmm sorted test (#2463)	2025-08-05 06:27:40 -07:00
test_random.py
test_reduce.py	fix complex reduce + nan propagation in min and max (#2377)	2025-07-15 18:19:47 -07:00
test_tree.py
test_upsample.py
test_vmap.py