7 Commits

Author	SHA1	Message	Date
Awni Hannun	da8c885784	Simplify removes no-ops from the tape (#1759) ... * simplify removes no-ops from the tape * comment	2025-01-09 11:23:19 -08:00
Awni Hannun	1ccaf80575	Dynamic broadcasting for shapeless compile/export (#1722) ... * working towards dynamic broadcast * shapeless broadcast * fix build + nits * use broadcast arrays in quantize matmul * some cleanup / consistency * mend * some comments * add vjp, jvp for broadcast axes	2025-01-09 11:04:24 -08:00
Jesper Stemann Andersen	c9c81d0584	Added additional missing unordered_map include that fixes build on FreeBSD (#1755)	2025-01-07 08:27:55 -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	69a2991614	allow compiling lambdas in C++ (#1650) ... * allow compiling lambdas in C++ * fix test * more tests * auto detect capture-less lambda	2024-12-06 13:13:21 -08:00
Awni Hannun	0ab8e099e8	Fix cpu segfault (#1488) ... * fix cpu segfault * nit in tests	2024-10-14 16:17:03 -07:00
Awni Hannun	7b463ffb07	Ios compile (#784) ... * try to fix build for ios * skip cpu compile * fix namespace * fix namespace * Use CMake for platform specific cpu compile --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>	2024-03-04 20:02:26 -08:00