Aashiq Dheeraj bb6565ef14 add fftshift and ifftshift fft helpers (#2135) ... * add fftshift and ifftshift fft helpers * address comments * axes have to be iterable * fix fp error in roll + add test --------- Co-authored-by: Aashiq Dheeraj <aashiq@aashiq-mbp-m4.local>		2025-04-29 22:13:45 -07:00
..
mlx_distributed_tests.py	Distributed layers (#1270)	2025-03-21 13:52:17 -07:00
mlx_tests.py	Make MLX build on x64 macOS (#901)	2024-03-27 06:14:29 -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	Fix the test and add custom min/max reductions for uncommon MPI types (#2060)	2025-04-10 17:01:17 -07:00
test_array.py	fix pinv (#2110)	2025-04-23 13:08:28 -07:00
test_autograd.py	Add stats and limit to common allocator and enable tests (#1988)	2025-03-21 12:28:36 -07:00
test_bf16.py	fix creating array from bf16 tensors in jax / torch (#1305)	2024-08-01 16:20:51 -07:00
test_blas.py	Gather mm new kernel and small refactoring (#2040)	2025-04-14 16:37:36 -07:00
test_compile.py	move memory APIs into top level mlx.core (#1982)	2025-03-21 07:25:12 -07:00
test_constants.py	Array api (#1289)	2024-07-26 10:40:49 -07:00
test_conv_transpose.py	Added output_padding parameters in conv_transpose (#2092)	2025-04-23 09:26:33 -07:00
test_conv.py	Depthwise Conv2D optimization (#2036)	2025-04-03 09:42:04 -07:00
test_device.py	Adds device context manager (#679)	2024-02-14 14:14:58 -08:00
test_double.py	Double for lapack (#1904)	2025-02-25 11:39:36 -08:00
test_einsum.py	Einsum ellipsis (#1788)	2025-01-25 01:28:03 -08:00
test_eval.py	Add stats and limit to common allocator and enable tests (#1988)	2025-03-21 12:28:36 -07:00
test_export_import.py	move memory APIs into top level mlx.core (#1982)	2025-03-21 07:25:12 -07:00
test_fast_sdpa.py	Add float mask to sdpa vector (#2068)	2025-04-11 17:29:40 -07:00
test_fast.py	RMS norm without scaling (#1915)	2025-02-28 20:26:57 -08:00
test_fft.py	add fftshift and ifftshift fft helpers (#2135)	2025-04-29 22:13:45 -07:00
test_graph.py	Multi output primitives (#330)	2024-01-08 16:39:08 -08:00
test_init.py	feat: Add orthogonal initializer and corresponding tests (#1651)	2025-01-13 07:29:20 -08:00
test_linalg.py	fix pinv (#2110)	2025-04-23 13:08:28 -07:00
test_load.py	Add stats and limit to common allocator and enable tests (#1988)	2025-03-21 12:28:36 -07:00
test_losses.py	Improve stability of BCE loss calculation for input probabilities close to or exactly 0 or 1 (#1280)	2024-07-24 08:38:22 -07:00
test_memory.py	Add stats and limit to common allocator and enable tests (#1988)	2025-03-21 12:28:36 -07:00
test_nn.py	loading empty list is ok when strict = false (#1834)	2025-02-05 16:19:27 -08:00
test_ops.py	add fftshift and ifftshift fft helpers (#2135)	2025-04-29 22:13:45 -07:00
test_optimizers.py	Add a multi optimizer (#1916)	2025-03-04 13:16:35 -08:00
test_quantized.py	Enable vjp for quantized scale and bias (#2129)	2025-04-29 13:03:09 -07:00
test_random.py	No reshapes in quantized embedding (#1682)	2024-12-09 18:57:38 -08:00
test_reduce.py	Reduce specializations (#1607)	2024-11-21 19:53:00 -08:00
test_tree.py	Add a multi optimizer (#1916)	2025-03-04 13:16:35 -08:00
test_upsample.py	Upsample with bicubic interpolation (#967)	2024-04-10 15:47:22 -07:00
test_vmap.py	conv vmap (#2102)	2025-04-21 13:04:39 -07:00