5fa1a8d59f
Support indexing with any type which implmented __index__ ( #3210 )
2026-03-19 15:19:08 -07:00
75f74ea9bc
Fix building with CUDA toolkit 13.2 ( #3273 )
2026-03-19 08:31:44 +09:00
7bc61cceed
Slice update with operation ( #3266 )
2026-03-18 06:18:02 -07:00
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
fcbdd05022
More useful error for large indices ( #3079 )
2026-01-29 13:02:39 -08:00
fd27829efa
Build and test python package on Windows CI ( #3049 )
2026-01-24 07:22:36 +09:00
edab937248
Add asarray to __array_namespace__ ( #2966 )
2026-01-12 06:16:27 -08:00
cacbdbf995
Fix init from double ( #2861 )
2025-12-03 06:08:11 -08:00
8879ee00eb
Support more Numpy interfaces for masked_scatter ( #2832 )
2025-12-01 17:51:02 -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
1c9ae1eaa1
cuda fix flaky test ( #2646 )
2025-10-02 15:40:04 -07:00
fbbf3b9b3e
Support pickling array for bfloat16 ( #2586 )
...
* add bfloat16 pickling
* Improvements
* improve
---------
Co-authored-by: Chen-Chen Yeh <ge96noj@mytum.de >
2025-09-22 20:12:15 -07:00
e7c6e1db82
no segfault with uninitialized array.at ( #2514 )
2025-08-18 08:33:38 -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
c35f4d089a
start cuda circle config ( #2256 )
...
* rebase
* fix metal kernel linking issue on cuda
* start cuda circle config
2025-06-10 21:19:47 -07:00
a2cadb8218
real and imag properties ( #2189 )
2025-05-15 18:17:50 -07:00
fbc89e3ced
fix pinv ( #2110 )
2025-04-23 13:08:28 -07:00
e9e268336b
LogCumSumExp ( #2069 )
2025-04-13 01:27:29 -07:00
5580b47291
iinfo and scalar overflow detection ( #2009 )
2025-03-27 19:54:56 -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
4c1dfa58b7
xor op on arrays ( #1875 )
2025-02-17 00:24:53 -08:00
55c5ac7820
fix int64 bug ( #1860 )
2025-02-12 19:23:46 -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
c3628eea49
Add mx.finfo and use it when making causal mask ( #1726 )
...
* finfo
* fixes
* docs
2024-12-19 14:52:41 -08:00
dfccd17ab9
Use psutil to get memory info on Windows ( #1700 )
2024-12-13 19:50:13 -08:00
bc2a29f033
fix ( #1654 )
2024-12-06 10:48:58 -08:00
bf481e8e5d
Fix sibling leak ( #1590 )
...
* add test
* fix + test
* fix fix
2024-11-18 19:17:01 -08:00
57c6aa7188
fix multi output leak ( #1548 )
2024-10-31 09:32:01 -07:00
d2ff04a4f2
fix format ( #1539 )
2024-10-28 18:29:14 -07:00
3d17077187
Add mx.array.__format__ ( #1521 )
...
* add __format__
* actually test something
* fix
2024-10-24 11:11:39 -07:00
f374b6ca4d
Bump nanobind to 2.2 ( #1461 )
...
* bump nanobind
* extension version for tests
2024-10-07 16:52:40 -07:00
0070e1db40
Fix deep recursion with siblings ( #1462 )
...
* fix recursion with siblings
* fix
* add test
* increase tol
2024-10-07 06:15:33 -07:00
11371fe251
Test to prevent bugs like #1386 ( #1391 )
...
* updated test_array for missing ops
* formatting changes
2024-09-04 17:24:30 -07:00
d1183821a7
int() and float() for mx.array ( #1360 )
2024-08-25 20:41:44 -07:00
7b456fd2c0
Array api ( #1289 )
...
* some updates for numpy 2.0 and array api
* some updates for numpy 2.0 and array api
* fix array api doc
2024-07-26 10:40:49 -07:00
bdb36c9a63
add zero vjps for bitwise ops and gather w.r.t. index ( #1256 )
2024-07-07 21:34:59 -07:00
ab977109db
feat: Added dlpack device ( #1165 )
...
* feat: Added dlpack device
* feat: Added device_id to dlpack device
* feat: Added device_id to dlpack device
* doc: updated conversion docs
* doc: updated numpy.rst dlpack information
* doc: updated numpy.rst dlpack information
* Update docs/src/usage/numpy.rst
* Update docs/src/usage/numpy.rst
---------
Co-authored-by: Venkat Ramnan Kalyanakumar <venkatramnankalyanakumar@Venkats-MacBook-Air.local >
Co-authored-by: Awni Hannun <awni.hannun@gmail.com >
2024-05-31 12:29:01 -07:00
eb8321d863
list based indexing ( #1150 )
2024-05-22 15:52:05 -07:00
fb71a82ada
Fix copy bug with many dims ( #1137 )
2024-05-17 21:10:03 -07:00
81dd33af66
allow conversion to dlpack ( #1120 )
2024-05-16 16:11:37 -07:00
631dfbe673
fix scatter index bug ( #1122 )
2024-05-14 15:04:58 -07:00
9814a2ae12
fix conversion to array ( #1070 )
2024-05-06 16:02:49 -07:00
7f7b9662ea
Fix leak for multi-output primitives which are never detached ( #1059 )
...
* fix multi output leak
* ignore arrays that will be detached
* add some comments
* stray print
2024-05-01 07:31:45 -07:00
09f1777896
fix slice update indexing ( #1053 )
2024-04-29 12:17:40 -07:00
490c0c4fdc
[Fix] expand axes for dimension with integer indices in mlx_slice_update ( #1035 )
...
* Not sure if this is correct
* Format
* Edit tests
* Add negative test
* Format
* add one more test
---------
Co-authored-by: Awni Hannun <awni@apple.com >
2024-04-29 07:57:28 -07:00
ef5f7d1aea
Fix buffer protocol buffer size designation ( #1010 )
2024-04-19 06:06:13 -07:00
0caf35f4b8
Better exceptions in case of invalid operations on mlx.core.array ( #910 ) ( #926 )
...
* Nicer exceptions for ops on non-arrays
2024-04-02 21:11:24 -07:00
639e06e1f3
Indexing bug fix ( #947 )
...
* Fix axes accounting
* Add tests
2024-04-01 12:18:50 -07:00
AN Long