mlx/mlx/io
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
..
CMakeLists.txt Chore: add pre-commit hook for cmake (#1362) 2024-09-16 12:53:01 -07:00
gguf_quants.cpp More shape type (#1705) 2024-12-19 08:08:20 -08:00
gguf.cpp More shape type (#1705) 2024-12-19 08:08:20 -08:00
gguf.h More shape type (#1705) 2024-12-19 08:08:20 -08:00
load.cpp More shape type (#1705) 2024-12-19 08:08:20 -08:00
load.h Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
no_gguf.cpp Add compiler flags to disable safetensors and gguf (#1098) 2024-05-09 17:39:44 -07:00
no_safetensors.cpp Add compiler flags to disable safetensors and gguf (#1098) 2024-05-09 17:39:44 -07:00
safetensors.cpp Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
threadpool.h Replace the use of result_of_t with invoke_result_t (#1397) 2024-09-06 19:52:57 -07:00