Skip to content

Pull requests: ml-explore/mlx

New pull request New
51 Open 2,352 Closed
51 Open 2,352 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

Add metadata to exported functions
#3855 opened Jul 16, 2026 by eyupcanakman Contributor Loading…
1
Fix incorrect nvfp4 quantized_matmul through the split-K path
#3854 opened Jul 16, 2026 by metascroy Loading…
Guard NAX MetalPerformancePrimitives include behind __has_include
#3853 opened Jul 16, 2026 by metascroy Loading…
Metal Cholesky
#3851 opened Jul 16, 2026 by kapellirohith • Draft
[CUDA][Improvement] RMSNorm forward speed up
#3850 opened Jul 16, 2026 by nastya236 Collaborator Loading…
Fix installed static MLX package on Windows
#3848 opened Jul 16, 2026 by yingjiacai Loading…
3 of 4 tasks
Fix log_cosh_loss docstring to document the element-wise loss
#3846 opened Jul 15, 2026 by winklemad Loading…
3 of 4 tasks
Fix conv_transpose maxBufferLength failures on Metal via tiled unfold
#3845 opened Jul 14, 2026 by eyupcanakman Contributor Loading…
3 of 4 tasks
Use unroll_count(4) for the NAX attention Q@K.T loop
#3843 opened Jul 13, 2026 by wyanzhao Loading…
4 tasks done
Add a fused full-attention path for head_dim 256 on NAX devices
#3842 opened Jul 13, 2026 by wyanzhao Loading…
4 tasks done
1
Extend SDPA vector 2-pass kernel to query sequence lengths up to 16
#3838 opened Jul 13, 2026 by katlun-lgtm Contributor Loading…
3 of 4 tasks
2
Fix eigh UPLO and zero-size eigh/svd on the CPU
#3834 opened Jul 11, 2026 by kapellirohith Loading…
3 of 4 tasks
Fix custom metal kernel cache collision for same name, different source (#3832)
#3833 opened Jul 11, 2026 by katlun-lgtm Contributor Loading…
1
Fix Glorot/He uniform init docstrings to label the uniform bound, not sigma
#3831 opened Jul 10, 2026 by vineethsaivs Loading…
2 of 4 tasks
1
Fix captured random state in compile
#3828 opened Jul 10, 2026 by angeloskath Member Loading…
1
5
Fixes an axis mismatch bug in matrix norm for case -1 and 1
#3827 opened Jul 9, 2026 by danlee2002 Contributor Loading…
4 tasks done
3
Make index autodiff errors explicitly recommend stop_gradient
#3820 opened Jul 8, 2026 by dogukanveziroglu Contributor Loading…
5
Refactor wheel building script
#3818 opened Jul 8, 2026 by zcbenz Collaborator Loading…
Fix BatchNorm running variance estimator
#3817 opened Jul 8, 2026 by ishtihoss Contributor Loading…
9
Fix shapeless matmul with dynamic batch dimensions
#3813 opened Jul 7, 2026 by varshneydevansh Contributor Loading…
3 of 4 tasks
1
3
Add builds for free-threaded python
#3812 opened Jul 7, 2026 by zcbenz Collaborator Loading…
1
[CUDA][Improvement] Fast path for gather
#3794 opened Jul 3, 2026 by nastya236 Collaborator Loading…
3
Raise qmv batch limit for large matrices on M5-class GPUs
#3791 opened Jul 2, 2026 by pierre427 Contributor Loading…
6
Decompose small kernel-depth 3D convs into 2D convs
#3785 opened Jul 1, 2026 by katlun-lgtm Contributor Loading…
1
5
[WIP] [CUDA] fsdp
#3768 opened Jun 25, 2026 by nastya236 Collaborator Loading…
ProTip! Adding no:label will show everything without a label.