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
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
