Skip to content

Pull requests: ml-explore/mlx

New pull request New
38 Open 2,284 Closed
38 Open 2,284 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 zero VJPs for scatter indices
#3790 opened Jul 2, 2026 by wgu9 Contributor Loading…
1
Fix kl_div_loss docstring formula and two doc nits
#3788 opened Jul 2, 2026 by Pablosinyores Contributor Loading…
Fix HardShrink to accept its documented lambd argument
#3786 opened Jul 1, 2026 by Pablosinyores Contributor Loading…
1
Decompose small kernel-depth 3D convs into 2D convs (#3625)
#3785 opened Jul 1, 2026 by katlun-lgtm Contributor Loading…
1
2
[WIP] [CUDA] fsdp
#3768 opened Jun 25, 2026 by nastya236 Collaborator • Draft
Fix fp_qmv_impl small-output-dim branch using raw fp8 scale byte
#3763 opened Jun 24, 2026 by jax-0n-git Loading…
1
1
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
2
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
Add math mode option for custom Metal kernels
#3728 opened Jun 19, 2026 by shubhxho Loading…
4 tasks done
1
3
Fix compiled kernel correctness for negative-strided inputs
#3720 opened Jun 19, 2026 by lyonsno Loading…
1
7
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 CPU gather transposing column-contiguous slices
#3647 opened Jun 9, 2026 by tillahoffmann Contributor Loading…
1
Add stft and istft to mlx.core.fft
#3639 opened Jun 8, 2026 by eyupcanakman Contributor Loading…
4 tasks done
6
Add reflect and symmetric padding modes to mx.pad
#3608 opened May 30, 2026 by katlun-lgtm Contributor Loading…
3
Implement batched matmul for large 1D dot products
#3580 opened May 22, 2026 by Ved235 Loading…
2 of 4 tasks
23
feat(fast): add sliding-window SDPA kernel path
#3552 opened May 15, 2026 by rabbitson87 Loading…
9
[CPU - Linux] AVX SIMD backend for fp16 and bf16 matmul
#3502 opened May 9, 2026 by acsweet Loading…
3 of 4 tasks
23
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
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
[Metal] Fused Flash Attention backward (VJP) kernels
#3241 opened Mar 11, 2026 by Brooooooklyn Contributor Loading…
1
Add all_to_all collective primitive
#3164 opened Feb 24, 2026 by 0xDaizz Loading…
2 of 4 tasks
1
ProTip! no:milestone will show everything without a milestone.