Skip to content

Pull requests: ml-explore/mlx

New pull request New
54 Open 2,267 Closed
54 Open 2,267 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

[WIP] [CUDA] fsdp
#3768 opened Jun 25, 2026 by nastya236 Collaborator • Draft
Fix complex vjps for several unary ops
#3766 opened Jun 25, 2026 by obchain Contributor Loading…
3 of 4 tasks
1
Add small-batch quantized matvec kernel (qmv_wide)
#3764 opened Jun 24, 2026 by jessegross Contributor Loading…
Fix fp_qmv_impl small-output-dim branch using raw fp8 scale byte
#3763 opened Jun 24, 2026 by jax-0n-git Loading…
1
Fix expm1 for complex inputs on the CPU backend
#3760 opened Jun 24, 2026 by obchain Contributor Loading…
3 of 4 tasks
1
Fix docstring nits
#3758 opened Jun 24, 2026 by ishtihoss Contributor Loading…
3
[CUDA] Add gather_qqmm
#3757 opened Jun 24, 2026 by zcbenz Collaborator Loading…
Fix infinite norm negative axis mismatch bug for matrices with 2 or more dimensions.
#3756 opened Jun 23, 2026 by danlee2002 Loading…
4 tasks done
1
Add vecdot to array API namespace
#3748 opened Jun 23, 2026 by wgu9 Contributor Loading…
1
Propagate CPU errors to events
#3742 opened Jun 22, 2026 by zcbenz Collaborator Loading…
8
Improve CPU float64 exp SIMD accuracy
#3740 opened Jun 22, 2026 by chrismicah Contributor • Draft
1
Fix CPU dynamic slice copy bound for collapsed shapes
#3739 opened Jun 21, 2026 by Lyxot Contributor Loading…
3 tasks done
4
array API: add cumulative_sum and cumulative_prod
#3731 opened Jun 20, 2026 by katlun-lgtm Contributor Loading…
5
array API: add positive, logical_xor, trunc, count_nonzero, diff, full_like
#3730 opened Jun 20, 2026 by katlun-lgtm Contributor Loading…
1
Add math mode option for custom Metal kernels
#3728 opened Jun 19, 2026 by shubhxho Loading…
4 tasks done
1
2
Fix compiled kernel correctness for negative-strided inputs
#3720 opened Jun 19, 2026 by lyonsno Loading…
1
7
Improves the MLX-vs-PyTorch benchmark runner for automate…
#3713 opened Jun 18, 2026 by aeiwz Loading…
3 of 4 tasks
1
Implement jvp for cumulative logsumexp
#3711 opened Jun 17, 2026 by obchain Contributor Loading…
3 of 4 tasks
1
2
Fix logsumexp jvp to reduce along the axis
#3708 opened Jun 17, 2026 by obchain Contributor Loading…
3 of 4 tasks
1
1
[CUDA] [Improvement] Rope without copy
#3704 opened Jun 16, 2026 by nastya236 Collaborator Loading…
4
Add metadata to exported functions
#3682 opened Jun 14, 2026 by Ritesh102000 Loading…
1
1
Add antialias support to nn.Upsample for linear and cubic modes
#3677 opened Jun 14, 2026 by lyonsno Loading…
Fix state corruption when a primitive throws during eval
#3675 opened Jun 13, 2026 by WindChimeRan Loading…
4 tasks done
2
Fused SDPA full-attention for head_dim=192/256 (revival of #3293)
#3660 opened Jun 11, 2026 by hojin12312 Loading…
3 of 4 tasks
1
Fix multi-wire recv prefill deadlock in jaccl ring backend
#3654 opened Jun 10, 2026 by jasonpaulso Loading…
2 of 4 tasks
1
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.