Awni Hannun e88f2d4a8e fix cross entropy axis param (#2641) ...

* fix cross entropy axis param

* faster grad clipping

2025-10-01 16:49:55 -07: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

Add mode parameter for quantization (#2499)

2025-08-28 06:45:26 -07:00

mlx_distributed_tests.py

Distributed layers (#1270)

2025-03-21 13:52:17 -07:00

mlx_tests.py

Detect cache thrashing in LRUCache (#2600)

2025-09-19 09:12:14 +09: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

nccl_test_distributed.py

Remove stream from average grads so it uses default (#2532)

2025-08-25 15:56:29 -07:00

ring_test_distributed.py

fix ring distributed test (#2380)

2025-07-16 11:25:24 -07:00

test_array.py

Support pickling array for bfloat16 (#2586)

2025-09-22 20:12:15 -07:00

test_autograd.py

Fix imag() vjp (#2367)

2025-07-14 13:11:16 -07: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

[CUDA] Set bias as input when using bias epilogue (#2584)

2025-09-11 15:31:09 +09:00

test_compile.py

Compile now can attach arbitrary data to an entry (#2634)

2025-09-30 13:33:27 -07: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

implement Convolution::output_shape (#2601)

2025-09-22 10:09:45 -07:00

test_fast_sdpa.py

avoid producing NaN in attention (#2608)

2025-09-22 13:10:43 -07:00

test_fast.py

Add batch offsets for mx.fast.rope (#2564)

2025-09-08 17:35:07 -07: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 python testing for cuda with ability to skip list of tests (#2295)

2025-06-15 10:56:48 -07:00

test_load.py

add load with path tests (#2543)

2025-08-26 14:24:47 -07:00

test_losses.py

fix cross entropy axis param (#2641)

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

test_memory.py

add python testing for cuda with ability to skip list of tests (#2295)

2025-06-15 10:56:48 -07:00

test_nn.py

Add mode parameter for quantization (#2499)

2025-08-28 06:45:26 -07:00

test_ops.py

expose depends (#2606)

2025-09-18 10:06:15 -07:00

test_optimizers.py

Adding support for the Muon Optimizer (#1914)

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

test_quantized.py

fix quantized vjp for mxfp4 (#2555)

2025-08-29 10:06:15 -07: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

fix complex reduce + nan propagation in min and max (#2377)

2025-07-15 18:19:47 -07:00

test_tree.py

add python testing for cuda with ability to skip list of tests (#2295)

2025-06-15 10:56:48 -07: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 python testing for cuda with ability to skip list of tests (#2295)

2025-06-15 10:56:48 -07:00