Commit Graph

256 Commits

Author SHA1 Message Date
Awni Hannun
d8f7b8cda6 docs update 2025-07-25 22:53:11 +00:00
Awni Hannun
8f68182d95 docs update 2025-07-25 22:53:11 +00:00
Awni Hannun
aa1647f94d docs update 2025-07-25 22:53:10 +00:00
Awni Hannun
0ec311dff3 docs update 2025-07-25 22:53:10 +00:00
Awni Hannun
85f70be0e6 docs update 2025-07-25 22:53:10 +00:00
Awni Hannun
d7a78fbe2b docs 2025-07-25 22:53:10 +00:00
Awni Hannun
20469ce2d1 docs update 2025-07-25 22:53:09 +00:00
Awni Hannun
93cf6e9f5c docs update 2025-07-25 22:53:09 +00:00
Awni Hannun
61c03b86a4 docs update 2025-07-25 22:53:09 +00:00
Awni Hannun
ba26fe5be5 docs update 2025-07-25 22:53:09 +00:00
Awni Hannun
8b5550f9f8 docs update 2025-07-25 22:53:09 +00:00
Awni Hannun
e1f3a7b14d use proper version 2025-07-25 22:53:08 +00:00
Awni Hannun
eede93197d docs update 2025-07-25 22:53:08 +00:00
Awni Hannun
fd78c54288 docs update 2025-07-25 22:53:08 +00:00
Awni Hannun
fdf4088123 docs update 2025-07-25 22:53:08 +00:00
Awni Hannun
9c44222630 docs update 2025-07-25 22:53:08 +00:00
Awni Hannun
f65f98fc82 remove uneeded files in docs 2025-07-25 22:53:08 +00:00
Awni Hannun
21cae9cb8f update docs 2025-07-25 22:53:08 +00:00
Awni Hannun
30ea2df988 docs update 2025-07-25 22:53:08 +00:00
Awni Hannun
d9d0777c2e docs up 2025-07-25 22:53:08 +00:00
Awni Hannun
d67cd9230c docs up 2025-07-25 22:53:07 +00:00
Awni Hannun
d03b91923e docs update 2025-07-25 22:53:07 +00:00
Awni Hannun
8bea0a4eb8 docs 2025-07-25 22:53:07 +00:00
Awni Hannun
0250e203f6 docs 2025-07-25 22:53:07 +00:00
Awni Hannun
f75712551d update docs 2025-07-25 22:53:07 +00:00
Awni Hannun
af2c3689fe docs 2025-07-25 22:53:07 +00:00
Awni Hannun
5ac2eec7b3 docs 2025-07-25 22:53:07 +00:00
Awni Hannun
f89de9c25d docs 2025-07-25 22:53:07 +00:00
Awni Hannun
9ad2650c9d docs 2025-07-25 22:53:07 +00:00
Awni Hannun
ea288788f8 docs 2025-07-25 22:53:07 +00:00
Awni Hannun
efe5c824af docs 2025-07-25 22:53:07 +00:00
Awni Hannun
e6ffce1a9b docs 2025-07-25 22:53:07 +00:00
Awni Hannun
a847d1dbd0 docs 2025-07-25 22:53:07 +00:00
Awni Hannun
dca1d17eb9 docs 2025-07-25 22:53:07 +00: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
Awni Hannun
c1eb9d05d9
non-symmetric eig and eigh (#2188) 2025-05-15 13:01:44 -07:00
Aashiq Dheeraj
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
Yury Popov
e9e268336b
LogCumSumExp (#2069) 2025-04-13 01:27:29 -07:00
Awni Hannun
9c6953bda7
Fix stubgen (#2065)
* Fix stubgen

* add multi optim to docs
2025-04-11 12:02:54 -07:00
Stephen Panaro
90823d2938
Add missing funcs to docs (#2021) 2025-03-30 18:29:33 -07:00
Awni Hannun
fe597e141c
add pinv to doc (#2020) 2025-03-30 15:54:18 -07:00