13 Commits

Author	SHA1	Message	Date
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	f110357aaa	Bump nanobind to 2.4 + fix (#1710) ... * bump nanobind to 2.4 + fix * fix	2024-12-17 10:57:54 -08:00
Cheng	0bf19037ca	Remove "using namespace mlx::core" in python/src (#1689)	2024-12-11 15:45:39 -08:00
Awni Hannun	40c62c1321	Use int64 stride everywhere (#1671) ... * use int64 stride everywhere * fix ext * fix ext * more shape + cleanup * one more * few more	2024-12-09 11:09:02 -08:00
Awni Hannun	020f048cd0	A few updates for CPU (#1482) ... * some updates * format * fix * nit	2024-10-14 12:45:49 -07:00
Alex Barron	d1183821a7	int() and float() for mx.array (#1360)	2024-08-25 20:41:44 -07:00
Awni Hannun	10b5835501	fix creating array from bf16 tensors in jax / torch (#1305)	2024-08-01 16:20:51 -07:00
Alex Barron	95d11bda06	Fix NumPy 2.0 pickle test (#1221) ... * fix numpy version <2 temporarily * typo * better fix * Fix just for bfloat16 --------- Co-authored-by: Alex Barron <abarron22@apple.com>	2024-06-23 05:47:22 -07:00
Awni Hannun	eb8321d863	list based indexing (#1150)	2024-05-22 15:52:05 -07:00
Awni Hannun	81dd33af66	allow conversion to dlpack (#1120)	2024-05-16 16:11:37 -07:00
Awni Hannun	8a0677d56d	Shared events for synchronization + async eval (#998) ... * more async eval * fix rebase * try correct async eval * fix async * more tests for async eval * use shared events for synchronization * comment + cleanup * with autorelease pool * fix no metal build * fix compile * fix patch * don't eval if asyn evale'd * don't use is_evaled * comments * more multi stream tests * try and cleanup use of is_evaled * use a status flag	2024-04-17 06:16:02 -07:00
Jack Mousseau	8e686764ac	Ensure shape dimensions are within supported integer range (#566) (#704) ... * Ensure shape dimensions are within supported integer range (#566) * fix build * fix rebase bug --------- Co-authored-by: Awni Hannun <awni@apple.com>	2024-03-25 13:29:45 -07:00
Awni Hannun	9a8ee00246	Switch to nanobind (#839) ... * mostly builds * most tests pass * fix circle build * add back buffer protocol * includes * fix for py38 * limit to cpu device * include * fix stubs * move signatures for docs * stubgen + docs fix * doc for compiled function, comments	2024-03-18 20:12:25 -07:00