-
Notifications
You must be signed in to change notification settings - Fork 1.7k
Pull requests: ml-explore/mlx
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Fix SDPA vmap with GQA/MQA shapes (n_heads != n_kv_heads)
#3385
opened Apr 8, 2026 by
Brooooooklyn
Loading…
4 tasks done
Add basic fp8/bfp8 support, future implementations of 8-bit floating …
#3374
opened Apr 4, 2026 by
Geramy
Loading…
3 tasks done
Fix int16 overflow in SDPA NAX mask indexing for KV sequences > 32K
#3361
opened Apr 3, 2026 by
Clydingus
Loading…
4 tasks done
Prevent out-of-bounds memory access caused by corrupt tensor.ndim in gguf file
#3359
opened Apr 3, 2026 by
MillaFleurs
Loading…
4 tasks done
Add TurboQuant KV cache compression with native Metal SDPA kernel
#3328
opened Mar 28, 2026 by
arozanov
Loading…
4 tasks done
[Metal] Fused Flash Attention backward (VJP) kernels
#3241
opened Mar 11, 2026 by
Brooooooklyn
Loading…
Fix shapeless compile eliding reductions on size-1 dimensions
#3202
opened Mar 4, 2026 by
ghstrider
Loading…
Add bessel_i0e and bessel_i1e ops
#3193
opened Mar 3, 2026 by
robert-johansson
Loading…
2 of 3 tasks
Fix command buffer memory tracking to use bytes instead of elements
#3192
opened Mar 3, 2026 by
hxu296
Loading…
3 of 4 tasks
Add 1-bit affine quantization support (Metal)
#3161
opened Feb 24, 2026 by
khosravipasha
Loading…
4 tasks done
feat: native i0 (modified Bessel function) and kaiser window
#3156
opened Feb 22, 2026 by
Vlor999
Loading…
4 tasks done
[JACCL] Fix silent data corruption from unchecked RDMA work completion status
#3152
opened Feb 21, 2026 by
0xDaizz
Loading…
[CUDA][Performance] Add radix select implementation for efficient partition operations
#3117
opened Feb 9, 2026 by
Lyxot
Loading…
3 of 4 tasks
Prevent allocator from operating on residency set during metal capture
#3114
opened Feb 8, 2026 by
robertmsale
•
Draft
3 of 4 tasks
Previous Next
ProTip!
Mix and match filters to narrow down what you’re looking for.