zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-29 13:01:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
4ba0c24a8f
mlx/mlx
History
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
..
3rdparty jagrit's commit files 2023-11-29 10:52:08 -08:00
backend Make mx.compile work on Windows (#1697) 2024-12-24 07:02:33 -08:00
distributed More shape type (#1705) 2024-12-19 08:08:20 -08:00
io Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
types Add mx.finfo and use it when making causal mask (#1726) 2024-12-19 14:52:41 -08:00
allocator.cpp Fix sibling leak (#1590) 2024-11-18 19:17:01 -08:00
allocator.h Allow querying the allocator for the buffer size (#1404) 2024-09-11 21:02:16 -07:00
array.cpp Explicit parentheses around some logical operators (#1732) 2024-12-24 07:02:20 -08:00
array.h More shape type (#1705) 2024-12-19 08:08:20 -08:00
CMakeLists.txt Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
compile_impl.h Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
compile.cpp Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
compile.h allow compiling lambdas in C++ (#1650) 2024-12-06 13:13:21 -08:00
device.cpp copyright + ack 2023-11-30 11:12:53 -08:00
device.h minor fixes (#1194) 2024-06-12 22:06:49 -07:00
dtype.cpp dtype is copy assignable (#1436) 2024-09-25 12:07:13 -07:00
dtype.h dtype is copy assignable (#1436) 2024-09-25 12:07:13 -07:00
einsum.cpp More shape type (#1705) 2024-12-19 08:08:20 -08:00
einsum.h Einsum (#1269) 2024-07-25 09:36:44 -07:00
event.h Fix array is_available race cases (#1468) 2024-10-07 19:13:50 -07:00
export_impl.h Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
export.cpp Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
export.h Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
fast_primitives.h Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
fast.cpp Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
fast.h Allow offset to be an mx.array for mx.fast.rope (#1724) 2024-12-19 15:51:44 -08:00
fft.cpp Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
fft.h Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
graph_utils.cpp Remove unnecessary string copies (#891) 2024-03-28 13:14:59 -07:00
graph_utils.h Added missing unordered_map includes (#1635) 2024-12-02 07:03:03 -08:00
io.h Added missing unordered_map includes (#1635) 2024-12-02 07:03:03 -08:00
linalg.cpp More shape type (#1705) 2024-12-19 08:08:20 -08:00
linalg.h Eigenvalues and eigenvectors (#1334) 2024-10-22 12:18:48 -07:00
mlx.h Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
ops.cpp Explicit parentheses around some logical operators (#1732) 2024-12-24 07:02:20 -08:00
ops.h More shape type (#1705) 2024-12-19 08:08:20 -08:00
primitives.cpp Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
primitives.h Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
random.cpp Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
random.h Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
scheduler.cpp Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
scheduler.h Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
stream.h Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
transforms_impl.h Fix cpu segfault (#1488) 2024-10-14 16:17:03 -07:00
transforms.cpp fix synch race (#1719) 2024-12-18 12:25:16 -08:00
transforms.h Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
utils.cpp Add mx.finfo and use it when making causal mask (#1726) 2024-12-19 14:52:41 -08:00
utils.h Add mx.finfo and use it when making causal mask (#1726) 2024-12-19 14:52:41 -08:00