Skip to content

Pull requests: ml-explore/mlx

New pull request New
47 Open 2,282 Closed
47 Open 2,282 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 BatchNorm docstring to cover 4D input and test the NHWC path
#3782 opened Jun 29, 2026 by Pablosinyores Contributor Loading…
Fix complex vjps for cos, sqrt, and rsqrt
#3781 opened Jun 29, 2026 by obchain Contributor Loading…
3 of 4 tasks
1
Fix complex vjps for the inverse trig and hyperbolic ops
#3779 opened Jun 29, 2026 by obchain Contributor Loading…
3 of 4 tasks
1
Add CI for Windows CUDA build
#3775 opened Jun 28, 2026 by zcbenz Collaborator Loading…
[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
Fix expm1 for complex inputs on the CPU backend
#3760 opened Jun 24, 2026 by obchain Contributor Loading…
3 of 4 tasks
1
1
[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
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
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
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
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
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.