mlx/python/tests
Awni Hannun 8402a2acf4
Fix complex power and print (#2286)
* fix complex power and print

* fix complex matmul shape
2025-06-13 11:13:00 -07:00
..
mlx_distributed_tests.py
mlx_tests.py
mpi_test_distributed.py
ring_test_distributed.py
test_array.py start cuda circle config (#2256) 2025-06-10 21:19:47 -07:00
test_autograd.py
test_bf16.py
test_blas.py Fix complex power and print (#2286) 2025-06-13 11:13:00 -07:00
test_compile.py
test_constants.py
test_conv_transpose.py
test_conv.py Add load_safe to the general conv loaders (#2258) 2025-06-10 20:58:16 -07:00
test_device.py start cuda circle config (#2256) 2025-06-10 21:19:47 -07:00
test_double.py
test_einsum.py
test_eval.py
test_export_import.py fix conv export (#2265) 2025-06-10 09:34:01 -07:00
test_fast_sdpa.py fix batched vector sdpa (#2152) 2025-05-05 13:13:03 -07:00
test_fast.py Fix unintuitive metal kernel caching (#2242) 2025-06-06 20:08:15 -07:00
test_fft.py Fix some complex vjps (#2178) 2025-05-14 23:37:12 -07:00
test_graph.py
test_init.py
test_linalg.py Refactor the lu test (#2250) 2025-06-07 06:12:08 -07:00
test_load.py
test_losses.py
test_memory.py
test_nn.py default strict mode for module update and update_modules (#2239) 2025-06-05 15:27:02 -07:00
test_ops.py Fix complex power and print (#2286) 2025-06-13 11:13:00 -07:00
test_optimizers.py start cuda circle config (#2256) 2025-06-10 21:19:47 -07:00
test_quantized.py 5bit quants (#2226) 2025-05-30 12:12:10 -07:00
test_random.py Add random normal distribution for complex numbers (#2182) 2025-05-13 22:43:45 -07:00
test_reduce.py
test_tree.py
test_upsample.py Nearest upsample (#2202) 2025-05-19 11:23:38 -07:00
test_vmap.py 5bit quants (#2226) 2025-05-30 12:12:10 -07:00