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
d6d9b24801
Conjugate VJP and JVP support ( #3386 )
2026-04-09 15:04:46 -07:00
7cd73c4202
[Metal] Support sorting complex numbers ( #3314 )
2026-04-01 12:40:50 +09:00
f8eda2c61b
[CUDA] support sorting complex numbers ( #3286 )
2026-03-25 12:35:02 +09:00
82809ebd12
Fix sort NaN handling for float16 and bfloat16 ( #3269 )
2026-03-19 15:19:41 -07:00
a25399cbd4
Validate num_splits in split ( #3234 )
...
Co-authored-by: KD2YCU <me@kd2ycu.com >
2026-03-09 21:47:08 -07:00
8cd377b7db
Add the bartlett function ( #3155 )
2026-03-03 11:40:54 -08:00
a8ba5ac3e0
Implement mlx.core.blackman ( #3136 )
2026-02-25 13:42:40 -08:00
360639c2df
Add the hamming window function ( #3135 )
2026-02-17 00:56:05 -08:00
3bbe87e6dc
Add hanning window function ( #3124 )
2026-02-16 09:44:49 -08:00
9bac6f8584
Allow take on empty array when it makes sense ( #3046 )
2026-01-23 07:25:46 -08:00
099dcc0f4c
Expose to/from fp8 in Python and don't auto-convert fp8 when loading from safetensors ( #2985 )
2026-01-13 15:48:21 -08:00
4a09264236
Tolerance for some ops tests on cuda ( #2815 )
2025-11-21 16:06:16 -08:00
b3825ac149
Add Masked Scatter ( #2663 )
...
Co-authored-by: Awni Hannun <awni@apple.com >
Co-authored-by: Angelos Katharopoulos <katharas@gmail.com >
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com >
2025-11-19 14:53:32 -08:00
66519fb348
fix slice ( #2758 )
2025-11-13 11:30:02 -08:00
539d8322d1
add median op ( #2705 )
2025-10-27 11:33:42 -07:00
9cbb1b0148
Modified sort behavior when running CPU or Metal to match NumPy/JAX ( #2667 )
...
* Modified sort behavior when running CPU or Metal to match NumPy/JAX sorting behavior.
* Modified sort behavior when running CPU or Metal to match NumPy/JAX
* nits
---------
Co-authored-by: Awni Hannun <awni@apple.com >
2025-10-13 14:36:45 -07:00
630350ad3e
Precise sigmoid ( #2659 )
...
* bump patch
* Sigmoid matches PyTorch and is more precise on tails
2025-10-10 10:05:23 -07:00
50cc09887f
expose depends ( #2606 )
2025-09-18 10:06:15 -07:00
7ef8a6f2d5
[CUDA] fix sort ( #2550 )
...
* [CUDA] fix sort
* fix test
2025-08-27 19:48:43 -07:00
e843c4d8d5
fix power ( #2523 )
2025-08-21 06:46:01 -07:00
ef631d63af
faster rms norm ( #2433 )
2025-07-29 13:12:00 -07:00
49114f28ab
fix flaky test ( #2371 )
2025-07-14 17:16:18 -07:00
c552ff2451
[CUDA] Fix back-end bugs and enable corresponding tests ( #2296 )
...
* Fix some cuda back-end bugs and enable corresponding tests
* more fixes
* enable more tests
* format
2025-06-16 08:45:40 -07:00
4fda5fbdf9
add python testing for cuda with ability to skip list of tests ( #2295 )
2025-06-15 10:56:48 -07:00
8402a2acf4
Fix complex power and print ( #2286 )
...
* fix complex power and print
* fix complex matmul shape
2025-06-13 11:13:00 -07:00
3aa9cf3f9e
Fix put_along_axis for empty arrays ( #2181 )
2025-05-13 14:27:53 -07:00
8f3d208dce
Close a couple edge case bugs: hadamard and addmm on empty inputs ( #2177 )
...
* handle hadamard and addmm on empty inputs
* fix
2025-05-12 10:48:57 -07:00
481349495b
GPU Hadamard for large N ( #1879 )
2025-05-01 17:19:17 -07:00
bb6565ef14
add fftshift and ifftshift fft helpers ( #2135 )
...
* add fftshift and ifftshift fft helpers
* address comments
* axes have to be iterable
* fix fp error in roll + add test
---------
Co-authored-by: Aashiq Dheeraj <aashiq@aashiq-mbp-m4.local >
2025-04-29 22:13:45 -07:00
3836445241
Add broadcast_shapes in python API ( #2091 )
2025-04-22 18:57:39 -07:00
1d2c9d6a07
Complex scan ( #2094 )
2025-04-22 18:56:28 -07:00
fdadc4f22c
Add more complex unary ops ( #2101 )
2025-04-21 13:04:54 -07:00
e9e268336b
LogCumSumExp ( #2069 )
2025-04-13 01:27:29 -07:00
de5f38fd48
Custom logsumexp ( #2028 )
...
* initial custom logsumexp
* more tests
* comments + fix
2025-03-31 07:36:55 -07:00
28f39e9038
Log for complex numbers in Metal ( #2025 )
...
* Log for complex numbers in Metal
* fix log2
2025-03-30 17:04:38 -07:00
2a980a76ce
Add stats and limit to common allocator and enable tests ( #1988 )
...
* add stats to common allocator and enable tests
* linux memory and default
* fix
2025-03-21 12:28:36 -07:00
4e1994e9d7
move memory APIs into top level mlx.core ( #1982 )
2025-03-21 07:25:12 -07:00
6bcd6bcf70
fix donation in scan ( #1917 )
2025-03-03 11:30:59 -08:00
4e7cd31d12
Fix slice data size ( #1913 )
...
* fix slice data size
* add test
2025-03-02 21:50:42 -08:00
2d0f384b6f
fix simd erf_inv ( #1896 )
2025-02-24 13:57:47 -08:00
1a2cb72030
Ensure linspace always contains start and stop ( #1883 )
2025-02-19 13:53:20 -08:00
5cd97f7ffe
Bitwise Inverse ( #1862 )
...
* add bitwise inverse
* add vmap + fix nojit
* inverse -> invert
* add to compile + remove unused
2025-02-13 08:44:14 -08:00
af1b725fda
Fix a couple of slicing bugs ( #1827 )
...
* fix a few bugs
* fix conv grad
* speedup test
* comment
2025-02-05 19:50:08 -08:00
9174606d4c
fix sort ( #1835 )
2025-02-05 17:16:27 -08:00
b7c9f1d38f
scatter axis + gather axis primitives ( #1813 )
...
* scatter axis + gather axis primitives
* add transforms
* comment
2025-01-31 20:48:08 -08:00
4758c8baa1
Start to cleanup/unify accelerate and common back-ends (Part 1/N) ( #1777 )
...
* start to cleanup/unify accelerate and common back-ends
* more progress
* simplify
* add half type and allow infs in simd exp
* unify softmax + quantized, more dispatches to simd quantized mm
* add sin/cos, use simd in vector-scalar ops
* faster CPU vectorize quant
* faster erf/erfinv
2025-01-29 14:34:49 -08:00
1ccaf80575
Dynamic broadcasting for shapeless compile/export ( #1722 )
...
* working towards dynamic broadcast
* shapeless broadcast
* fix build + nits
* use broadcast arrays in quantize matmul
* some cleanup / consistency
* mend
* some comments
* add vjp, jvp for broadcast axes
2025-01-09 11:04:24 -08:00
516ded618b
Dynamic slicing ( #1741 )
...
* dynamic slice and slice update
* python bindings + tests + fix set item
* fix compile issue
* comment
* fix jit
2025-01-07 14:02:16 -08:00
259025100e
Fix nd ternary on GPU ( #1746 )
2025-01-03 11:52:17 -08:00
serenposh