Skip to content

Pull requests: ml-explore/mlx

New pull request New
49 Open 1,986 Closed
49 Open 1,986 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Slice update with operation
#3266 opened Mar 16, 2026 by angeloskath Loading…
Merge DeviceStream into CommandEncoder
#3264 opened Mar 16, 2026 by zcbenz Loading…
Add output_shapes for AddMM, Slice, and CustomKernel
#3262 opened Mar 16, 2026 by pHequals7 Loading…
5 tasks
add causal_upper_left mask option to scaled_dot_product_attention
#3254 opened Mar 14, 2026 by mm65x • Draft
4 tasks done
feat: implement Group.split() for JACCL and Ring backends
#3245 opened Mar 11, 2026 by machiabeli Loading…
[Metal] Fused Flash Attention backward (VJP) kernels
#3241 opened Mar 11, 2026 by Brooooooklyn Loading…
1
Add bias support to QQLinear
#3215 opened Mar 6, 2026 by mdepree Loading…
4 tasks done
5
Support indexing with any type which implmented __index__
#3210 opened Mar 5, 2026 by aisk Loading…
3 of 4 tasks
1
Fix shapeless compile eliding reductions on size-1 dimensions
#3202 opened Mar 4, 2026 by ghstrider Loading…
1
2
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
2
Add lgamma and digamma ops
#3181 opened Feb 27, 2026 by robert-johansson Loading…
3 of 4 tasks
1
Add all_to_all collective primitive
#3164 opened Feb 24, 2026 by 0xDaizz Loading…
2 of 4 tasks
1
Add Expert Parallelism for MoE inference
#3158 opened Feb 23, 2026 by 0xDaizz • Draft
1 of 7 tasks
3
[WIP] columnwise quantize with tma
#3157 opened Feb 23, 2026 by nastya236 • Draft
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…
6
Fix np bfloat16 misinterpreted as complex
#3146 opened Feb 19, 2026 by kellen-sun Loading…
3 of 4 tasks
1
2
Add recurrent SNN docs, tests, and benchmark scaffold
#3140 opened Feb 18, 2026 by Ikaikaalika Loading…
[Metal][Performance]: Add split-K for quantized matmul (small M)
#3120 opened Feb 12, 2026 by Ziqiao-git Loading…
3 of 4 tasks
16
[CUDA][Performance] Add radix select implementation for efficient partition operations
#3117 opened Feb 9, 2026 by Lyxot Loading…
3 of 4 tasks
20
Prevent allocator from operating on residency set during metal capture
#3114 opened Feb 8, 2026 by robertmsale • Draft
3 of 4 tasks
6
PR 3103 Updating Documentation
#3104 opened Feb 6, 2026 by MillaFleurs Loading…
4 tasks done
5
Proposal to add dynamic backends to MLX
#3098 opened Feb 5, 2026 by dhiltgen Loading…
1 of 4 tasks
2
Add bounds checking on GPU
#3091 opened Feb 2, 2026 by nhey • Draft
3 of 4 tasks
2
ProTip! Adding no:label will show everything without a label.