zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-15 21:21:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
1,262 Commits 37 Branches 66 Tags 765 MiB
f64d446626
Commit Graph

244 Commits

Author SHA1 Message Date
Awni Hannun
f64d446626 docs update 2025-07-08 21:30:19 +00:00
Awni Hannun
7f95802cc5 docs update 2025-07-08 21:30:19 +00:00
Awni Hannun
f320e5bd1f docs update 2025-07-08 21:30:18 +00:00
Awni Hannun
3f0b33ecd5 use proper version 2025-07-08 21:30:18 +00:00
Awni Hannun
a88103a649 docs update 2025-07-08 21:30:18 +00:00
Awni Hannun
d0c2c3d1ca docs update 2025-07-08 21:30:18 +00:00
Awni Hannun
4c160546aa docs update 2025-07-08 21:30:18 +00:00
Awni Hannun
829d18cae7 docs update 2025-07-08 21:30:18 +00:00
Awni Hannun
798eab9f5e remove uneeded files in docs 2025-07-08 21:30:18 +00:00
Awni Hannun
3c372a8f6f update docs 2025-07-08 21:30:18 +00:00
Awni Hannun
2010d815aa docs update 2025-07-08 21:30:17 +00:00
Awni Hannun
51399f9a37 docs up 2025-07-08 21:30:17 +00:00
Awni Hannun
de0fc12d34 docs up 2025-07-08 21:30:17 +00:00
Awni Hannun
37fdd687b0 docs update 2025-07-08 21:30:17 +00:00
Awni Hannun
6d35eb1742 docs 2025-07-08 21:30:17 +00:00
Awni Hannun
ffe51a69ca docs 2025-07-08 21:30:17 +00:00
Awni Hannun
087651b1b1 update docs 2025-07-08 21:30:17 +00:00
Awni Hannun
efda283748 docs 2025-07-08 21:30:17 +00:00
Awni Hannun
1e6c2e6cfe docs 2025-07-08 21:30:17 +00:00
Awni Hannun
cb7a2041a5 docs 2025-07-08 21:30:17 +00:00
Awni Hannun
41bb77f7e3 docs 2025-07-08 21:30:17 +00:00
Awni Hannun
14d44f5869 docs 2025-07-08 21:30:17 +00:00
Awni Hannun
dc37663f6b docs 2025-07-08 21:30:17 +00:00
Awni Hannun
527f486e59 docs 2025-07-08 21:30:16 +00:00
Awni Hannun
96b48b3154 docs 2025-07-08 21:30:16 +00:00
Awni Hannun
075ca7feed docs 2025-07-08 21:30:16 +00: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
Abe Leininger
a5ededf1c3
CPU LU factorization and linear solvers (#1451) 
* linalg solve backend

* nits

* more nits + fix

* luf primitive and lu, solve, and solve_triangular backends

* changes / nits

---------

Co-authored-by: Awni Hannun <awni@apple.com>
 2025-02-10 12:32:24 -08:00
Awni Hannun
1c0c118f7c
Fp64 on the CPU (#1843) 
* add fp64 data type

* clean build

* update docs

* fix bug
 2025-02-07 15:52:22 -08:00
Awni Hannun
7064fed1b1
Minor update on MPI docs (#1805) 2025-01-28 11:00:08 -08:00
Awni Hannun
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
Awni Hannun
b51d70a83c
export docs (#1747) 2025-01-03 15:04:17 -08:00