Awni Hannun
c098245664
docs update
2025-06-04 01:01:49 +00:00
Awni Hannun
981bf7ae2b
docs update
2025-06-04 01:01:49 +00:00
Awni Hannun
ce95a29690
docs update
2025-06-04 01:01:49 +00:00
Awni Hannun
9e69a72b8c
docs update
2025-06-04 01:01:48 +00:00
Awni Hannun
17470bf630
remove uneeded files in docs
2025-06-04 01:01:48 +00:00
Awni Hannun
a693e6e1d8
update docs
2025-06-04 01:01:48 +00:00
Awni Hannun
fd34610634
docs update
2025-06-04 01:01:48 +00:00
Awni Hannun
84bebc2161
docs up
2025-06-04 01:01:48 +00:00
Awni Hannun
9882295582
docs up
2025-06-04 01:01:48 +00:00
Awni Hannun
ebd913400a
docs update
2025-06-04 01:01:48 +00:00
Awni Hannun
217cdf3fc9
docs
2025-06-04 01:01:48 +00:00
Awni Hannun
43cd655ba1
docs
2025-06-04 01:01:48 +00:00
Awni Hannun
8c406bcb9b
update docs
2025-06-04 01:01:48 +00:00
Awni Hannun
01489e172d
docs
2025-06-04 01:01:48 +00:00
Awni Hannun
616449e363
docs
2025-06-04 01:01:48 +00:00
Awni Hannun
a66e6d3214
docs
2025-06-04 01:01:48 +00:00
Awni Hannun
d3d0ad9564
docs
2025-06-04 01:01:47 +00:00
Awni Hannun
a60a600c6a
docs
2025-06-04 01:01:47 +00:00
Awni Hannun
e84ebcf0b9
docs
2025-06-04 01:01:47 +00:00
Awni Hannun
372f2ac025
docs
2025-06-04 01:01:47 +00:00
Awni Hannun
80322b562e
docs
2025-06-04 01:01:47 +00:00
Awni Hannun
fbd10a48d4
docs
2025-06-04 01:01:47 +00: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
Awni Hannun
c9d30aa6ac
MLX in C++ example ( #1736 )
...
* MLX in C++ example
* nits
* fix docs
2025-01-02 19:09:04 -08:00
Awni Hannun
ae69cb15e9
shapeless compile in docs and partially shapeless reshape ( #1742 )
2025-01-02 16:24:42 -08:00
Danilo Peixoto
92ec632ad5
Fix Distributed Communication documentation ( #1731 )
...
* Add missing `size()` method call for group
2025-01-02 14:08:38 -08:00
Awni Hannun
4ba0c24a8f
Export / import functions to / from a file ( #1642 )
...
* export and import functions
* refactor + works for few primitives
* nit
* allow primitives with state
* nit
* nit
* simplify serialize / deserialize
* fix for constants
* python bindings
* maybe fix serialize failure case
* add example
* more primitives, training kind of works
* same result for python and c++
* some fixes
* fix export
* template it up
* some simplificatoin
* rebase
* allow kwargs and multiple functions
* exporter
* more primitives for exporting
* deal with endianness
* handle invalid stream
* add docstring
2024-12-24 11:19:13 -08:00
Awni Hannun
c3628eea49
Add mx.finfo
and use it when making causal mask ( #1726 )
...
* finfo
* fixes
* docs
2024-12-19 14:52:41 -08:00
Cheng
f9640e049d
Install mlx.dll into the same dir with python bindings on Windows ( #1690 )
...
* Install mlx.dll into the same dir with python bindings on Windows
* Set BUILD_SHARED_LIBS for dlfcn-win32
* Update cmake requirements to 3.25
* Fix cmake style
2024-12-13 19:50:39 -08:00
Awni Hannun
4e1e9520e1
Flatten and unflatten ( #1692 )
...
* flatten and unflatten
* fix grad
* fix shape infer
* use squeeze + unsqueeze in get_item
2024-12-11 21:51:37 -08:00
Awni Hannun
40c62c1321
Use int64 stride everywhere ( #1671 )
...
* use int64 stride everywhere
* fix ext
* fix ext
* more shape + cleanup
* one more
* few more
2024-12-09 11:09:02 -08:00