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

..

allocator_tests.cpp

Metal validation (#432)

2024-01-11 11:57:24 -08:00

arg_reduce_tests.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

array_tests.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

autograd_tests.cpp

Fix x86 tests (#1691)

2024-12-11 07:47:18 -08:00

blas_tests.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

CMakeLists.txt

Export / import functions to / from a file (#1642)

2024-12-24 11:19:13 -08:00

compile_tests.cpp

Using math defines requires switch in MSVC (#1665)

2024-12-08 08:16:28 -08:00

creations_tests.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

custom_vjp_tests.cpp

Update pre-commit hooks (#984)

2024-04-11 07:27:53 -07:00

device_tests.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

einsum_tests.cpp

Einsum (#1269)

2024-07-25 09:36:44 -07:00

eval_tests.cpp

Shared events for synchronization + async eval (#998)

2024-04-17 06:16:02 -07:00

export_import_tests.cpp

Export / import functions to / from a file (#1642)

2024-12-24 11:19:13 -08:00

fft_tests.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

linalg_tests.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

load_tests.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

metal_tests.cpp

Expose function to clear memory cache (#1032)

2024-04-24 16:48:51 -07:00

ops_tests.cpp

More shape type (#1705)

2024-12-19 08:08:20 -08:00

random_tests.cpp

More shape type (#1705)

2024-12-19 08:08:20 -08:00

scheduler_tests.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

tests.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

utils_tests.cpp

Add mx.finfo and use it when making causal mask (#1726)

2024-12-19 14:52:41 -08:00

vmap_tests.cpp

More shape type (#1705)

2024-12-19 08:08:20 -08:00