zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-22 01:21:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
1,295 Commits 37 Branches 67 Tags 818 MiB
c620a28b16
Commit Graph

249 Commits

Author SHA1 Message Date
Awni Hannun
c620a28b16 docs update 2025-07-18 22:25:33 +00:00
Awni Hannun
3e724a7c98 docs update 2025-07-18 22:25:33 +00:00
Awni Hannun
f946f689a6 docs update 2025-07-18 22:25:32 +00:00
Awni Hannun
f77d99b285 docs update 2025-07-18 22:25:32 +00:00
Awni Hannun
1d2cadbc78 docs update 2025-07-18 22:25:32 +00:00
Awni Hannun
1b03079c4a use proper version 2025-07-18 22:25:32 +00:00
Awni Hannun
bc695f2050 docs update 2025-07-18 22:25:32 +00:00
Awni Hannun
1fbcfa159f docs update 2025-07-18 22:25:32 +00:00
Awni Hannun
2c714de62a docs update 2025-07-18 22:25:32 +00:00
Awni Hannun
e492638dff docs update 2025-07-18 22:25:31 +00:00
Awni Hannun
06b6d9a4d4 remove uneeded files in docs 2025-07-18 22:25:31 +00:00
Awni Hannun
f12615680d update docs 2025-07-18 22:25:31 +00:00
Awni Hannun
c465c51cbb docs update 2025-07-18 22:25:31 +00:00
Awni Hannun
7534da7269 docs up 2025-07-18 22:25:31 +00:00
Awni Hannun
2aeb6df29c docs up 2025-07-18 22:25:31 +00:00
Awni Hannun
f1dfa257d2 docs update 2025-07-18 22:25:31 +00:00
Awni Hannun
de4f3e72fd docs 2025-07-18 22:25:31 +00:00
Awni Hannun
9d7133097f docs 2025-07-18 22:25:31 +00:00
Awni Hannun
65b0bdffe1 update docs 2025-07-18 22:25:31 +00:00
Awni Hannun
6f38552812 docs 2025-07-18 22:25:31 +00:00
Awni Hannun
5aa62e5553 docs 2025-07-18 22:25:31 +00:00
Awni Hannun
e4d33acace docs 2025-07-18 22:25:30 +00:00
Awni Hannun
901a4ba68a docs 2025-07-18 22:25:30 +00:00
Awni Hannun
8fdf710d38 docs 2025-07-18 22:25:30 +00:00
Awni Hannun
86cae0f191 docs 2025-07-18 22:25:30 +00:00
Awni Hannun
1a9bcc0ef1 docs 2025-07-18 22:25:30 +00:00
Awni Hannun
07f0207405 docs 2025-07-18 22:25:30 +00:00
Awni Hannun
24b9da7d61 docs 2025-07-18 22:25:30 +00: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
Yi Wang
0da8506552
Update docs for extensions (#2004) 2025-03-25 18:35:03 -07:00
Awni Hannun
cfd7237a80
fix docs (#1991) 2025-03-21 19:58:53 -07:00
Awni Hannun
4e1994e9d7
move memory APIs into top level mlx.core (#1982) 2025-03-21 07:25:12 -07:00
Awni Hannun
7b7e2352cd
fix malloc or wait deadlock (#1976) 2025-03-20 16:48:43 -07:00
Awni Hannun
c4230747a1
redesign for faster cpu/gpu synch (#1869) 
* redesign for faster cpu/gpu synch

* load + more async CPU

* use command encoder API and move more ops to use it

* make fence back-end generic + CPU only fence

* faster build

* fix async eval

* fixes + handle temporaries

* fix / improve cpu conv

* remove unused status, fix siblings

* fix extensions

* fix

* fix no cpu build

* format

* comments

* fix perf regression, remove unecessary abort

* fix events, task limit cpu

* fix waiting

* fix donation / temporaries in normalization
 2025-03-06 19:23:38 -08:00
Angelos Katharopoulos
5d68082881
Ring docs (#1829) 2025-02-28 11:34:21 -08:00
Alex Barron
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