Tarjei Mandt ca14d3d835 Fix sharding of quantized models with non-power-of-2 bits (#3006)

2026-01-18 07:21:56 -08:00

..

__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

Expose to/from fp8 in Python and don't auto-convert fp8 when loading from safetensors (#2985)

2026-01-13 15:48:21 -08:00

mlx_distributed_tests.py

Nccl reduce scatter, all gather (#2727)

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

mlx_tests.py

Detect cache thrashing in LRUCache (#2600)

2025-09-19 09:12:14 +09:00

mpi_test_distributed.py

Nccl reduce scatter, all gather (#2727)

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

nccl_test_distributed.py

Nccl reduce scatter, all gather (#2727)

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

ring_test_distributed.py

Nccl reduce scatter, all gather (#2727)

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

test_array.py

Add asarray to __array_namespace__ (#2966)

2026-01-12 06:16:27 -08:00

test_autograd.py

Fix grad in place updates (#2899)

2025-12-11 14:44:58 -08: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

patch bump for next release (#2991)

2026-01-14 08:46:09 -08:00

test_compile.py

Fix input buffer donation in compile (#2897)

2025-12-11 06:07:03 -08: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

Fix overflow in large filter small channels (#2520)

2025-08-20 08:03:29 -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

Fix export scatters (#2852)

2025-12-02 11:24:40 -08:00

test_fast_sdpa.py

[CUDA] Support array mask in SDPA (#2822)

2025-11-26 11:08:58 +09:00

test_fast.py

Layer norm throws on dimension mismatch (#2870)

2025-12-04 11:21:05 -08: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 float64 Eig and complex64 SVD/Eig support (Fixes #2708) (#2737)

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

test_load.py

Expose to/from fp8 in Python and don't auto-convert fp8 when loading from safetensors (#2985)

2026-01-13 15:48:21 -08:00

test_losses.py

fix cross entropy axis param (#2641)

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

test_memory.py

fix memory count bug (#2717)

2025-10-30 14:27:15 -07:00

test_nn.py

Fix sharding of quantized models with non-power-of-2 bits (#3006)

2026-01-18 07:21:56 -08:00

test_ops.py

Expose to/from fp8 in Python and don't auto-convert fp8 when loading from safetensors (#2985)

2026-01-13 15:48:21 -08:00

test_optimizers.py

Adding support for the Muon Optimizer (#1914)

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

test_quantized.py

Metal/CPU nvfp4 and mxfp8 (#2946)

2025-12-22 20:45:19 -08: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_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

Fix grad in place updates (#2899)

2025-12-11 14:44:58 -08:00