jhavukainen 8c7bc30ce4 Align mlx::core::min op nan propagation with NumPy (#2346)

2025-07-10 06:20:43 -07:00

..

blas

…

comparative

[CUDA] Fix reductions (#2314)

2025-06-27 12:59:20 -07:00

batch_matmul_bench.py

…

compile_bench.py

…

conv1d_bench.py

…

conv2d_bench_cpu.py

Conv cpu improvements (#1410)

2024-09-15 18:45:10 -07:00

conv2d_train_bench_cpu.py

Conv cpu improvements (#1410)

2024-09-15 18:45:10 -07:00

conv2d_transpose_bench_cpu.py

Conv cpu improvements (#1410)

2024-09-15 18:45:10 -07:00

conv3d_bench_cpu.py

Conv cpu improvements (#1410)

2024-09-15 18:45:10 -07:00

conv3d_train_bench_cpu.py

Conv cpu improvements (#1410)

2024-09-15 18:45:10 -07:00

conv3d_transpose_bench_cpu.py

Conv cpu improvements (#1410)

2024-09-15 18:45:10 -07:00

conv_bench.py

…

conv_transpose_bench.py

…

conv_unaligned_bench.py

Add load_safe to the general conv loaders (#2258)

2025-06-10 20:58:16 -07:00

distributed_bench.py

…

einsum_bench.py

…

fft_bench.py

…

gather_bench.py

Remove unused modules (#1949)

2025-03-10 06:05:26 -07:00

gather_mm_bench.py

Gather qmm batched kernel and refactoring of quantized (#2078)

2025-04-17 13:53:11 -07:00

gather_qmm_bench.py

Gather qmm batched kernel and refactoring of quantized (#2078)

2025-04-17 13:53:11 -07:00

hadamard_bench.py

…

layer_norm_bench.py

Change layernorms to two pass algorithm (#2246)

2025-06-06 13:34:56 -07:00

rms_norm_bench.py

RMS norm without scaling (#1915)

2025-02-28 20:26:57 -08:00

rope_bench.py

…

scatter_bench.py

improvements to scatter / gather (#1541)

2024-10-30 19:30:54 -07:00

sdpa_bench.py

Support fused masking in Attention (#1924)

2025-03-20 11:01:32 -07:00

sdpa_vector_bench.py

Allow different value dimensions in sdpa_vector (#1811)

2025-01-31 20:58:59 -08:00

single_ops.py

Align mlx::core::min op nan propagation with NumPy (#2346)

2025-07-10 06:20:43 -07:00

synchronize_bench.py

Faster synchronization Fence primitive (#1773)

2025-01-17 18:42:19 -08:00

time_utils.py

…