mlx/python/tests at jaccl-examples - mlx - L'Électron Rare

__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

[CUDA] Handle residue k in qmm_naive (#3379 )

2026-04-18 13:30:07 +09:00

mlx_distributed_tests.py

Hybrid sharding (#3194 )

2026-03-10 11:47:25 +01:00

mlx_tests.py

Add clear_streams API for cleanup before exit (#3395 )

2026-04-14 18:41:32 +09:00

mpi_test_distributed.py

Nccl reduce scatter, all gather (#2727 )

2025-11-05 08:21:11 -08:00

nccl_test_distributed.py

Hybrid sharding (#3194 )

2026-03-10 11:47:25 +01:00

ring_test_distributed.py

Nccl reduce scatter, all gather (#2727 )

2025-11-05 08:21:11 -08:00

test_array.py

Add printoptions (#3333 )

2026-04-01 22:24:48 -07:00

test_autograd.py

Fix mx.prod vjp for complex types (#3433 )

2026-04-21 17:35:20 -07:00

test_bf16.py

Fix np bfloat16 misinterpreted as complex (#3146 )

2026-03-31 08:04:55 +09:00

test_blas.py

Fix flaky macOS test (#3063 )

2026-01-25 16:40:57 -08:00

test_compile.py

Add output_shapes for AddMM (#3262 )

2026-03-25 10:46:15 +09: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

Fixes for CUDA CI (#3413 )

2026-04-14 23:52:52 -07:00

test_device.py

GPU discovery (#3055 )

2026-01-26 09:54:13 -08:00

test_double.py

Use higher precision for linspace with double (#3029 )

2026-01-21 06:20:50 -08: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

Export: preserve Dtype state values in export callback arguments (#3145 )

2026-02-19 08:07:28 -08:00

test_fast_sdpa.py

Fixes for CUDA CI (#3413 )

2026-04-14 23:52:52 -07:00

test_fast.py

Validate dims in rope (#3230 )

2026-03-09 21:48:45 -07:00

test_fft.py

Add fftfreq, rfftfreq and scalar axes for fftshift/ifftshift (#3298 )

2026-03-31 18:29:16 -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 float64 Eig and complex64 SVD/Eig support (Fixes #2708 ) (#2737 )

2025-11-22 06:51:36 -08:00

test_load.py

Build and test python package on Windows CI (#3049 )

2026-01-24 07:22:36 +09:00

test_losses.py

fix cross entropy axis param (#2641 )

2025-10-01 16:49:55 -07:00

test_memory.py

GPU discovery (#3055 )

2026-01-26 09:54:13 -08:00

test_nn.py

Fix: Correct cross-attention query routing in Post-LN TransformerDecoderLayer (#3382 )

2026-04-07 09:16:12 -07:00

test_ops.py

Conjugate VJP and JVP support (#3386 )

2026-04-09 15:04:46 -07:00

test_optimizers.py

Adding support for the Muon Optimizer (#1914 )

2025-07-18 12:25:28 -07:00

test_quantized.py

Tensor scale nvfp4 (#3022 )

2026-02-18 11:19:26 +01: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

Add a 2-pass col reduce for CUDA (#2863 )

2025-12-04 15:53:59 -08:00

test_threads.py

ThreadLocalStream in C++ (#3405 )

2026-04-15 15:46:11 -07:00

test_tree.py

Added support for pytree types that inherit from tuple and typing.namedtuple (#2845 )

2025-12-04 11:06:45 -08: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 vmap for BroadcastAxes (#3344 )

2026-03-31 17:08:56 -07:00