Skip to content

Pull requests: ml-explore/mlx

New pull request New
46 Open 2,154 Closed
46 Open 2,154 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

Fix complex VJPs for log and exp
#3605 opened May 29, 2026 by CameronChurchwell Contributor Loading…
4 tasks done
Fix signed-integer overflow (UB) in roll and tile shape arithmetic
#3604 opened May 29, 2026 by devYRPauli Loading…
1
Fix CUDA all-reduce planning for large inputs
#3603 opened May 28, 2026 by sofinvalery Contributor Loading…
1
3
block_fp8: 2D-block FP8 quantized matmul + MoE kernels (DeepSeek-V3 / MiMo)
#3600 opened May 28, 2026 by yohann-bearzi Loading…
1
Add VJP for cumulative max and min scans
#3599 opened May 28, 2026 by devin-lai Loading…
2
Add metal::set_metallib_path() to override the metallib search path
#3597 opened May 28, 2026 by olilarkin Loading…
3
Cache JIT-compiled CUDA kernels
#3587 opened May 24, 2026 by zcbenz Collaborator Loading…
Implement batched matmul for large 1D dot products
#3580 opened May 22, 2026 by Ved235 Loading…
2 of 4 tasks
7
Add new_thread_unsafe_stream API
#3578 opened May 22, 2026 by zcbenz Collaborator Loading…
[CUDA] JIT-compile qmm_naive
#3576 opened May 21, 2026 by zcbenz Collaborator Loading…
Fix CPU GatherMM one-row NaNs
#3563 opened May 19, 2026 by TyceHerrman Loading…
1
5
Find SwiftPM metallib bundle near test binary
#3562 opened May 19, 2026 by RNT56 Loading…
1
feat(fast): add sliding-window SDPA kernel path
#3552 opened May 15, 2026 by rabbitson87 Loading…
9
Add Metal DLPack zero-copy sharing
#3531 opened May 11, 2026 by XXXXRT666 Contributor Loading…
4 tasks done
138
Catch error in CommandBuffer and poison the events
#3523 opened May 11, 2026 by zcbenz Collaborator Loading…
1
9
[CPU - Linux] AVX SIMD backend for fp16 and bf16 matmul
#3502 opened May 9, 2026 by acsweet Loading…
3 of 4 tasks
18
Use uv in macOS CI
#3491 opened May 7, 2026 by zcbenz Collaborator Loading…
2
Fix JACCL GID selection on Apple Thunderbolt RDMA (errno 22 RTR fail)
#3468 opened Apr 29, 2026 by danielkristofik Loading…
3 of 4 tasks
1
Reduce peak memory of large half-precision random.uniform/normal
#3439 opened Apr 22, 2026 by dogukanveziroglu Contributor • Draft
4 tasks done
5
fix: avoid clearing in-flight pipeline states in custom kernel cache
#3434 opened Apr 21, 2026 by Ziqiao-git Contributor Loading…
2
fsdp: improve state saving determinism and tree reconstruction
#3407 opened Apr 13, 2026 by coder-jayp Loading…
1 of 4 tasks
3
Fix SDPA vmap with GQA/MQA shapes (n_heads != n_kv_heads)
#3385 opened Apr 8, 2026 by Brooooooklyn Contributor Loading…
4 tasks done
1
Add basic fp8/bfp8 support, future implementations of 8-bit floating …
#3374 opened Apr 4, 2026 by Geramy Loading…
3 tasks done
2
[Metal] Fused Flash Attention backward (VJP) kernels
#3241 opened Mar 11, 2026 by Brooooooklyn Contributor Loading…
1
Add bessel_i0e and bessel_i1e ops
#3193 opened Mar 3, 2026 by robert-johansson Contributor Loading…
2 of 3 tasks
ProTip! Exclude everything labeled bug with -label:bug.