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
Cheng
ed4ec81bca
Link python extension with mlx statically on Windows ( #1716 )
...
* Link python extension with mlx statically on Windows
* More readable code
2024-12-18 19:26:04 -08:00
Awni Hannun
f110357aaa
Bump nanobind to 2.4 + fix ( #1710 )
...
* bump nanobind to 2.4 + fix
* fix
2024-12-17 10:57:54 -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
Cheng
310ad8d9db
Build OpenBLAS from source code for MSVC ( #1674 )
...
* Download OpenBLAS binaries when building with MSVC
* Download dlfcn-win32
* Link with dlfcn-win32 correctly
* Build OpenBLAS from source code
* Link with openblas statically
* Link with BLAS privately
2024-12-10 16:14:44 -08:00
Amethyst Shen
9df8fed046
Metal-cpp version bump ( #1668 )
...
* Metal-cpp version bump
Apple has released the stable version of Metal-cpp for macOS 15 and iOS 18. CMakeLists.txt is updated to build with it instead of the beta one.
* Fix style with cmake-format
2024-12-09 19:40:35 -08:00
Awni Hannun
50fa705125
patch bump ( #1656 )
2024-12-06 13:16:19 -08:00
Jesper Stemann Andersen
974bb54ab2
CMake: Enabled using Accelerate on x86_64 / x64 ( #1625 )
...
* CMake: Enabled using Accelerate on x86_64 / x64
Cf. https://github.com/JuliaPackaging/Yggdrasil/pull/9761
* CMake: Removed superfluous MLX_BUILD_ARM
2024-11-28 10:55:45 -08:00
Jesper Stemann Andersen
698e63a608
CMake: Build with dlfcn-win32 to have dlopen etc. on win32 ( #1628 )
...
Cf. https://github.com/JuliaPackaging/Yggdrasil/pull/9761
2024-11-27 13:14:13 -08:00
Awni Hannun
bb303c45a5
version ( #1617 )
2024-11-22 12:00:03 -08:00
Awni Hannun
2af7e8a9a6
fix cmake version ( #1601 )
2024-11-19 08:45:05 -08:00
Awni Hannun
9d7fa6b8e6
Use osx deployment target to pick Metal version ( #1595 )
...
* choose metal based on deployment target rather than system version
* nit
* unused compile def
2024-11-18 19:16:49 -08:00
Alex Barron
726dbd9267
v0.20.0 ( #1565 )
2024-11-05 12:37:57 -08:00
Awni Hannun
eac961ddb1
patch ( #1550 )
2024-10-31 16:10:14 -07:00
Awni Hannun
cde5b4ad80
patch ( #1546 )
2024-10-30 19:31:22 -07:00
Awni Hannun
35e9c87ab9
patch bump ( #1528 )
2024-10-25 13:13:23 -07:00
Awni Hannun
58a855682c
v0.19.0 ( #1502 )
2024-10-18 11:55:18 -07:00
Awni Hannun
c21331d47f
version bump ( #1477 )
2024-10-10 13:05:17 -07:00
Awni Hannun
3274c6a087
Fix array is_available race cases ( #1468 )
2024-10-07 19:13:50 -07:00
Cheng
5900e3249f
Fix building on Linux ( #1446 )
2024-09-30 07:00:39 -07:00
Awni Hannun
b1e2b53c2d
bump ( #1445 )
2024-09-27 13:53:02 -07:00
jjuang-apple
8d68a3e805
remove fmt dependencies from MLX install ( #1417 )
2024-09-16 13:32:28 -07:00
jjuang-apple
6bbcc453ef
avoid using find_library to make install truly portable ( #1416 )
2024-09-16 13:21:32 -07:00
Nripesh Niketan
669c27140d
Chore: add pre-commit hook for cmake ( #1362 )
...
* reset and lint
* format
---------
Co-authored-by: Awni Hannun <awni@apple.com>
2024-09-16 12:53:01 -07:00
Angelos Katharopoulos
d0c58841d1
Patch bump ( #1408 )
2024-09-12 16:44:23 -07:00
Angelos Katharopoulos
1600092e92
Patch bump ( #1376 )
2024-08-29 16:54:30 -07:00
Angelos Katharopoulos
8081df79be
Fix boolean all reduce bug ( #1355 )
2024-08-24 10:09:32 -07:00
Alex Barron
684e11c664
patch ( #1347 )
2024-08-23 10:42:02 -07:00
Awni Hannun
1086dc4db0
patch ( #1320 )
2024-08-12 16:13:33 -07:00
Angelos Katharopoulos
780c197f95
Fix test tolerance and patch bump ( #1315 )
2024-08-08 14:51:09 -07:00
Awni Hannun
e9e53856d2
patch bump ( #1287 )
2024-07-25 11:42:09 -07:00
Awni Hannun
e2aa6ec8ae
some fixes ( #1281 )
2024-07-23 11:49:05 -07:00
Alex Barron
d0da74209b
version bump ( #1260 )
2024-07-11 11:17:55 -07:00
Awni Hannun
d6383a1c6a
version bump ( #1239 )
2024-06-27 10:43:13 -07:00
Awni Hannun
56c8a33439
Get metal version from xcode ( #1228 )
...
* get metal version from xcode
* typo
* fix
2024-06-26 07:02:11 -07:00
Awni Hannun
af9079cc1f
version bump ( #1212 )
2024-06-14 11:28:51 -07:00
Awni Hannun
31e134be35
Build for macOS 15 ( #1208 )
...
* Build for macos 15
* metal32 as well
* comment
---------
Co-authored-by: Awni Hannun <Awni Hannun>
2024-06-13 13:31:44 -07:00
Awni Hannun
e84ba8056d
only allow openmpi ( #1209 )
2024-06-13 12:14:44 -07:00
Awni Hannun
cf236fc390
version ( #1191 )
2024-06-06 17:16:40 -07:00
Awni Hannun
079882495d
version bump ( #1172 )
2024-05-31 12:29:12 -07:00
Awni Hannun
9f9cb7a2ef
version bump ( #1154 )
2024-05-23 18:08:08 -07:00
Angelos Katharopoulos
50dfb664db
Comms ( #1097 )
...
* Start the communications branch using MPI
* Add ops and primitives
* Add python bindings for distributed
2024-05-23 17:04:02 -07:00
Awni Hannun
226748b3e7
JIT compile option for binary minimization ( #1091 )
...
* try cpp 20 for compile
* unary, binary, ternary in jit
* nits
* fix gather/scatter
* fix rebase
* reorg compile
* add ternary to compile
* jit copy
* jit compile flag
* fix build
* use linked function for ternary
* some nits
* docs + circle min size build
* docs + circle min size build
* fix extension
* fix no cpu build
* improve includes
2024-05-22 12:57:13 -07:00
Awni Hannun
6a9b584f3d
patch bump ( #1131 )
2024-05-16 20:51:33 -07:00
Awni Hannun
1873ffda01
Detect metal version and propagate correctly for JIT ( #1109 )
...
* detect metal version and propagate correctly for JIT
* remove softmax
* fix versions
2024-05-15 17:42:09 -07:00
Awni Hannun
7178ac0111
No CPU option for binary minimization ( #1105 )
...
* no cpu build option
* docs
* fix
2024-05-13 16:08:11 -07:00
Awni Hannun
8bd6bfa4b5
version ( #1099 )
2024-05-09 17:52:39 -07:00
Awni Hannun
8b1906abd0
Add compiler flags to disable safetensors and gguf ( #1098 )
...
* with docs
* nit
2024-05-09 17:39:44 -07:00
Nripesh Niketan
79c859e2e0
feat: implement clip_grad_norm
( #1043 )
...
* feat: implement `clip_grad_norm`
* pre-commit
* Add test for clip_grad_norm function in test_optimizers.py
* small fixes
* fix
* lint
* Update tree_reduce
* Update python/mlx/utils.py
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
* Update python/mlx/utils.py
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
* Update python/mlx/utils.py
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
* Update python/mlx/utils.py
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
* Update python/mlx/utils.py
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
* Update python/mlx/utils.py
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
* Refactor clip_grad_norm function to include documentation and improve readability
* format docstring
* Add acknowlegements
* text wrap
* pre-commit
* nits in docs
---------
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
Co-authored-by: Awni Hannun <awni@apple.com>
2024-05-03 09:07:02 -07:00
Awni Hannun
02a9fc7bfa
Patch bump ( #1067 )
...
* version
* use 0.12.2
2024-05-02 16:37:31 -07:00