Commit Graph

262 Commits

Author SHA1 Message Date
Awni Hannun
8d5c671135 docs up 2025-08-07 07:46:33 +00:00
Awni Hannun
60abe38be8 docs update 2025-08-07 07:46:33 +00:00
Awni Hannun
36427f6126 docs update 2025-08-07 07:46:33 +00:00
Awni Hannun
6c7354535e docs update 2025-08-07 07:46:33 +00:00
Awni Hannun
e595324804 docs update 2025-08-07 07:46:33 +00:00
Awni Hannun
185a879b05 docs update 2025-08-07 07:46:32 +00:00
Awni Hannun
3df9dab38c docs update 2025-08-07 07:46:32 +00:00
Awni Hannun
b86da5711b docs 2025-08-07 07:46:32 +00:00
Awni Hannun
2988d6c14d docs update 2025-08-07 07:46:32 +00:00
Awni Hannun
680a57bb33 docs update 2025-08-07 07:46:32 +00:00
Awni Hannun
b70128ce73 docs update 2025-08-07 07:46:31 +00:00
Awni Hannun
1364ef2259 docs update 2025-08-07 07:46:31 +00:00
Awni Hannun
2c0a1a3952 docs update 2025-08-07 07:46:31 +00:00
Awni Hannun
aea164fab8 use proper version 2025-08-07 07:46:31 +00:00
Awni Hannun
9abc99cfe9 docs update 2025-08-07 07:46:31 +00:00
Awni Hannun
4490e181db docs update 2025-08-07 07:46:31 +00:00
Awni Hannun
e04f658611 docs update 2025-08-07 07:46:31 +00:00
Awni Hannun
b04ef9ec37 docs update 2025-08-07 07:46:31 +00:00
Awni Hannun
f2365d5b12 remove uneeded files in docs 2025-08-07 07:46:30 +00:00
Awni Hannun
eea1a5f6ec update docs 2025-08-07 07:46:30 +00:00
Awni Hannun
1c7ebac536 docs update 2025-08-07 07:46:30 +00:00
Awni Hannun
c07de29458 docs up 2025-08-07 07:46:30 +00:00
Awni Hannun
c0dc02b8f4 docs up 2025-08-07 07:46:30 +00:00
Awni Hannun
e9e7ff09dd docs update 2025-08-07 07:46:30 +00:00
Awni Hannun
b024ed7744 docs 2025-08-07 07:46:30 +00:00
Awni Hannun
beda33e1a9 docs 2025-08-07 07:46:30 +00:00
Awni Hannun
83d006dfdd update docs 2025-08-07 07:46:30 +00:00
Awni Hannun
b906475af3 docs 2025-08-07 07:46:30 +00:00
Awni Hannun
0bb6edbe54 docs 2025-08-07 07:46:30 +00:00
Awni Hannun
b728f1fd66 docs 2025-08-07 07:46:30 +00:00
Awni Hannun
3bf234ff88 docs 2025-08-07 07:46:30 +00:00
Awni Hannun
082e939fc7 docs 2025-08-07 07:46:30 +00:00
Awni Hannun
75b55ffa19 docs 2025-08-07 07:46:29 +00:00
Awni Hannun
b8cf6bd778 docs 2025-08-07 07:46:29 +00:00
Awni Hannun
3e96be74e6 docs 2025-08-07 07:46:29 +00:00
Awni Hannun
312037d0f7 docs 2025-08-07 07:46:29 +00:00
Luca Vivona
728d4db582
Support destination arg in tree flatten/unflatten (#2450) 2025-08-06 15:34:59 -07:00
Awni Hannun
db5c7efcf6
revert default cuda install (#2465)
* revert default cuda install

* revert default cuda install
2025-08-06 06:19:12 -07:00
Awni Hannun
6ad0889c8a
default install cuda on linux (#2462) 2025-08-04 15:33:05 -07:00
Awni Hannun
da5912e4f2
fix custom metal extension (#2446) 2025-07-31 06:25:36 -07:00
Awni Hannun
dcb8319f3d
update install docs and requirements (#2419) 2025-07-25 12:13:19 -07:00
Gökdeniz Gülmez
deee214a95
Adding support for the Muon Optimizer (#1914)
* initial commit with workong optmimizer

* update ACKNOWLEDGMENTS.md

* nits and adding it to test

* nits

* G.astype(mx.bfloat16) to G.astype(G.dtype)

* G.ndim >= 2 to assert G.ndim == 2

* remove coments

* replace with  mx.addmm

* remove comments

* format

* nits

* match muon

* fix addmm

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2025-07-18 12:25:28 -07:00
Awni Hannun
f0a0b077a0
Install linux with mlx[cuda] and mlx[cpu] (#2356)
* install linux with mlx[cuda] and mlx[cpu]

* temp for testing

* cleanup circle, fix cuda repair

* update circle

* update circle

* decouple python bindings from core libraries
2025-07-14 17:17:33 -07:00
Cheng
d34f887abc
Add Primitive::name and remove Primitive::print (#2365) 2025-07-14 14:06:35 -07:00
Cheng
19facd4b20
Build with all cpu cores by default (#2336) 2025-07-07 06:06:45 -07:00
Awni Hannun
76831ed83d
Build CUDA release in Circle (#2306)
* cuda release

* add license
2025-06-19 15:26:36 -07:00
Awni Hannun
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
Awni Hannun
1ca616844b
Fix unintuitive metal kernel caching (#2242)
* Fix unintuitive metal kernel caching

* alternative solution
2025-06-06 20:08:15 -07:00
Awni Hannun
55b4062dd8
copyright in docs (#2214) 2025-05-21 17:13:04 -07:00
Awni Hannun
a2cadb8218
real and imag properties (#2189) 2025-05-15 18:17:50 -07:00