7 Commits

Author	SHA1	Message	Date
Awni Hannun	4fda5fbdf9	add python testing for cuda with ability to skip list of tests (#2295)	2025-06-15 10:56:48 -07:00
Awni Hannun	62fecf3e13	fix conv export (#2265)	2025-06-10 09:34:01 -07:00
Awni Hannun	9ce77798b1	fix export to work with gather/scatter axis (#2263)	2025-06-09 20:37:27 -07:00
Awni Hannun	9daa6b003f	fix shapeless export (#2148)	2025-05-01 15:02:02 -07:00
Awni Hannun	4e1994e9d7	move memory APIs into top level mlx.core (#1982)	2025-03-21 07:25:12 -07:00
Awni Hannun	2a45056ba8	Cycle leak break (#1856) ... * detect and break leaks in custom function * detect and break leaks in custom function	2025-02-11 14:45:02 -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