 Angelos Katharopoulos
|
ff57d875ea
|
Fix indexing bug in slice update with op (#3483)
|
2026-05-06 17:04:50 -07:00 |
|
|
Cheng
|
80bcd1c658
|
[CUDA] Fix half type matmul in cutlass kernels (#3469)
|
2026-05-06 08:35:53 +09:00 |
|
|
serenposh
|
1fdd4e23c2
|
Clearer error when shape dimension overflows int32 (#3425)
Co-authored-by: Kanishk <kanishk.chores@gmail.com>
Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
|
2026-05-05 09:53:36 +09:00 |
|
|
Abhilash Shankarampeta
|
0938db7e54
|
Add determinant and sign-log-determinant functions to mlx.core.linalg (#3416)
Co-authored-by: Lucas Fernandes Martins <Lucas-Fernandes-Martins@users.noreply.github.com>
|
2026-05-05 09:06:23 +09:00 |
|
|
Cameron Churchwell
|
68cf2fddd8
|
Fix mx.prod vjp for complex types (#3433)
|
2026-04-21 17:35:20 -07:00 |
|
|
Angelos Katharopoulos
|
705c828feb
|
Fix synchronize for ThreadLocalStream (#3429)
|
2026-04-20 11:29:49 -07:00 |
|
|
Cheng
|
fa4320d5fa
|
[CUDA] Handle residue k in qmm_naive (#3379)
|
2026-04-18 13:30:07 +09:00 |
|
|
Cheng
|
d142de6a20
|
[CUDA] gather_mm (#3414)
|
2026-04-17 16:53:44 +09:00 |
|
|
Cheng
|
dec6b4d10f
|
ThreadLocalStream in C++ (#3405)
|
2026-04-15 15:46:11 -07:00 |
|
|
NeuralNoble
|
fd8e849e26
|
Document sort stability and NaN handling (#3400)
|
2026-04-15 14:32:42 -07:00 |
|
|
Cheng
|
57bcced8cb
|
Fixes for CUDA CI (#3413)
|
2026-04-14 23:52:52 -07:00 |
|
|
Cheng
|
435f0b6cdb
|
Add clear_streams API for cleanup before exit (#3395)
|
2026-04-14 18:41:32 +09:00 |
|
|
Clydingus
|
a33b791615
|
Fix int16 overflow in SDPA NAX mask indexing for KV sequences > 32K (#3361)
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
|
2026-04-10 00:01:47 -07:00 |
|
|
Cameron Churchwell
|
d6d9b24801
|
Conjugate VJP and JVP support (#3386)
|
2026-04-09 15:04:46 -07:00 |
|
|
Shantanu Suryawanshi
|
a8776b7bbd
|
Fix: Correct cross-attention query routing in Post-LN TransformerDecoderLayer (#3382)
|
2026-04-07 09:16:12 -07:00 |
|
|
Angelos Katharopoulos
|
6a9a121d09
|
Add a convenience for making local streams in python (#3355)
|
2026-04-02 18:43:02 -07:00 |
|
|
Christophe Prat
|
befe42d303
|
Add printoptions (#3333)
|
2026-04-01 22:24:48 -07:00 |
|
|
Angelos Katharopoulos
|
b0748ad8de
|
Fix regression in array creation (#3353)
|
2026-04-01 11:30:36 -07:00 |
|
|
Cheng
|
2ffafe07f4
|
[CUDA] 3/5/6-bit quants for qmm_naive (#3352)
|
2026-04-01 20:13:01 +09:00 |
|
|
Cheng
|
1c9ee2f655
|
[CUDA] Fallback QMM (#3315)
|
2026-04-01 12:41:26 +09:00 |
|
|
Long Yixing
|
7cd73c4202
|
[Metal] Support sorting complex numbers (#3314)
|
2026-04-01 12:40:50 +09:00 |
|
|
declanhealy2
|
2105df91da
|
Add fftfreq, rfftfreq and scalar axes for fftshift/ifftshift (#3298)
|
2026-03-31 18:29:16 -07:00 |
|
|
Angelos Katharopoulos
|
1944cf67a2
|
Add vmap for BroadcastAxes (#3344)
|
2026-03-31 17:08:56 -07:00 |
|
|
Kellen Sun
|
8a6d28713c
|
Fix np bfloat16 misinterpreted as complex (#3146)
Co-authored-by: Cheng <git@zcbenz.com>
|
2026-03-31 08:04:55 +09:00 |
|
|
Long Yixing
|
0ff1115a46
|
[CUDA] Implement BlockMaskedMM (#3299)
Co-authored-by: Cheng <git@zcbenz.com>
|
2026-03-27 06:57:26 +09:00 |
|
|
Cheng
|
df7f7db943
|
Make each thread have its own default stream (#3281)
|
2026-03-25 15:48:49 +09:00 |
|
|
Sheldon Aristide
|
57c813f042
|
Add norm parameter to FFT transforms (#3287)
Co-authored-by: Cheng <git@zcbenz.com>
|
2026-03-25 13:27:40 +09:00 |
|
|
Long Yixing
|
f8eda2c61b
|
[CUDA] support sorting complex numbers (#3286)
|
2026-03-25 12:35:02 +09:00 |
|
|
Pranav Hari
|
bd200d6267
|
Add output_shapes for AddMM (#3262)
Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
|
2026-03-25 10:46:15 +09:00 |
|
|
Cheng
|
d01b83dfe7
|
Use nb::ndarray for checking arrays (#3283)
|
2026-03-25 10:44:54 +09:00 |
|
|
Sheldon Aristide
|
1b1c56352a
|
Fix moved-from shape bug in broadcast_arrays causing vmap bus error (#3310)
|
2026-03-24 17:02:31 -07:00 |
|
|
Sheldon Aristide
|
81530c261b
|
Implement Pad::vmap (#3304)
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
|
2026-03-24 15:02:31 -07:00 |
|
|
Cheng
|
70a0da6fca
|
Use thread local storage for frontend compile cache (#3280)
|
2026-03-20 07:44:45 +09:00 |
|
|
Long Yixing
|
82809ebd12
|
Fix sort NaN handling for float16 and bfloat16 (#3269)
|
2026-03-19 15:19:41 -07:00 |
|
|
AN Long
|
5fa1a8d59f
|
Support indexing with any type which implmented __index__ (#3210)
|
2026-03-19 15:19:08 -07:00 |
|
|
Cheng
|
21c11fc9b0
|
Create default random key lazily (#3278)
|
2026-03-19 20:22:52 +09:00 |
|
|
Angelos Katharopoulos
|
45af0df90b
|
Fix repr of conv layers (#3275)
|
2026-03-18 22:47:38 -07:00 |
|
|
Cheng
|
75f74ea9bc
|
Fix building with CUDA toolkit 13.2 (#3273)
|
2026-03-19 08:31:44 +09:00 |
|
|
Angelos Katharopoulos
|
7bc61cceed
|
Slice update with operation (#3266)
|
2026-03-18 06:18:02 -07:00 |
|
|
Ihar Hrachyshka
|
e353be8235
|
tests: harden memory leak check in test_siblings_without_eval (#3088)
Signed-off-by: Ihar Hrachyshka <ihar.hrachyshka@gmail.com>
|
2026-03-17 16:03:10 +09:00 |
|
|
mm65x
|
f226eeec9e
|
Fix nn.GRU skipping bhn bias when hidden is None (#3252)
Co-authored-by: mm65x <mm65x@users.noreply.github.com>
|
2026-03-16 13:28:14 -07:00 |
|
|
mm65x
|
505fc9850d
|
Fix comparison op JVP returning bool tangents instead of input dtype (#3253)
|
2026-03-16 10:57:28 -07:00 |
|
|
Long Yixing
|
0bdbfdb838
|
[CUDA] Implement MaskedScatter (#3151)
|
2026-03-15 10:33:55 +09:00 |
|
|
Lucas Newman
|
5d1700493a
|
[CUDA] Add FFT support (#3243)
|
2026-03-14 21:02:19 +09:00 |
|
|
Long Yixing
|
a9573f92f6
|
[CUDA] Implement SegmentedMM (#3238)
|
2026-03-11 13:31:43 -07:00 |
|
|
Anastasiia Filippova
|
e1e1399e1b
|
Hybrid sharding (#3194)
|
2026-03-10 11:47:25 +01:00 |
|
|
Cheng
|
9d03a1b0d9
|
[CUDA] Support 3/5/6-bit quants in QMV (#3236)
|
2026-03-10 19:09:48 +09:00 |
|
|
Michelle DiMarco
|
d2702a4fc1
|
Fix non-strict module update with extra weights (#3214)
Co-authored-by: Michelle DiMarco <m_dimarco@apple.com>
|
2026-03-09 22:03:50 -07:00 |
|
|
Dan Anderson
|
572e0a4ac3
|
Validate dims in rope (#3230)
Co-authored-by: KD2YCU <me@kd2ycu.com>
|
2026-03-09 21:48:45 -07:00 |
|
|
Dan Anderson
|
a25399cbd4
|
Validate num_splits in split (#3234)
Co-authored-by: KD2YCU <me@kd2ycu.com>
|
2026-03-09 21:47:08 -07:00 |
|