Skip to content

Pull requests: ml-explore/mlx

New pull request New
30 Open 1,879 Closed
30 Open 1,879 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 failing python tests on Windows
#3076 opened Jan 28, 2026 by zcbenz Loading…
Fallback to pinned host memory when managed memory is not supported
#3075 opened Jan 28, 2026 by zcbenz Loading…
Faster vector SDPA in CUDA
#3073 opened Jan 27, 2026 by awni • Draft
Fix ALiBi slopes for non-power-of-2 num_heads
#3071 opened Jan 27, 2026 by vovw Loading…
2 of 4 tasks
2
[Metal] [Performance] Add radix select implementation for efficient partition operations
#3069 opened Jan 27, 2026 by NripeshN Loading…
26
Fix double precision for exp, sin, cos, erf, erfinv on CPU
#3058 opened Jan 24, 2026 by Anri-Lombard Loading…
4 tasks done
1
5
[WIP] Quantized SDPA
#3026 opened Jan 20, 2026 by CC-Yeh • Draft
4 of 7 tasks
11
Tensor scale nvfp4
#3022 opened Jan 20, 2026 by nastya236 Loading…
1
Add support for x86 SIMD (SSE, AVX2, AVX512)
#3019 opened Jan 19, 2026 by dhiltgen • Draft
2 of 4 tasks
5
Fix edge checking in attention mask
#3002 opened Jan 15, 2026 by jagrit06 Loading…
4 tasks done
1
Fix FFT failure for array sizes 2^21 and 2^22
#2997 opened Jan 14, 2026 by Anri-Lombard Loading…
1
[Metal] Add Flash Attention VJP for training
#2995 opened Jan 14, 2026 by Brooooooklyn Loading…
2 tasks done
32
[WIP] Remove conda from setup
#2975 opened Jan 8, 2026 by angeloskath • Draft
2
[Docs] Simple example of using MLX distributed
#2973 opened Jan 8, 2026 by stefpi Loading…
7
WIP Windows CUDA support
#2972 opened Jan 7, 2026 by dhiltgen • Draft
2 of 4 tasks
10
Use lower-right causal mask alignment consistently
#2967 opened Jan 4, 2026 by Anri-Lombard Loading…
23
View-only layout ops mover
#2920 opened Dec 17, 2025 by CC-Yeh • Draft
3 of 4 tasks
1
load metallib not only as resource but in root of a bundle
#2885 opened Dec 9, 2025 by gin66 Loading…
4 tasks
2
[Feature] Additional padding modes
#2862 opened Dec 3, 2025 by sidtuladhar Loading…
4 tasks done
2
Fix memory leak in transforms (#2841)
#2853 opened Dec 2, 2025 by harsh-sutariya Loading…
4 tasks done
[WIP] Add mlx.core.searchsorted
#2817 opened Nov 21, 2025 by ppaleja • Draft
4 tasks done
17
Kernel for sparse matrix-dense matrix multiplication
#2796 opened Nov 19, 2025 by mercush Loading…
3 of 4 tasks
1
Prefer signed types, and remove -Wall -Wextra warnings
#2723 opened Oct 31, 2025 by andresy Loading…
4 tasks done
4
Add xielu activation function
#2603 opened Sep 17, 2025 by Goekdeniz-Guelmez • Draft
3 of 4 tasks
[Experiment] ROCm backend
#2300 opened Jun 16, 2025 by NripeshN Loading…
1
45
ProTip! Adding no:label will show everything without a label.