145 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Awni Hannun	172e696540	remove message from test output	2023-12-22 17:03:48 -08:00
dc-dc-dc	c6d7702ef0	remove pathlib refs	2023-12-22 19:06:31 -05:00
dc-dc-dc	ee6ce00aee	docs and made retain_graph optional bool	2023-12-22 16:19:31 -05:00
dc-dc-dc	fa093967ec	moved to subdirectory	2023-12-22 16:05:07 -05:00
dc-dc-dc	13fd7bd721	small fixes	2023-12-22 01:25:42 -05:00
dc-dc-dc	fdf9d99f0f	add back retain_graph argument	2023-12-22 01:17:08 -05:00
dc-dc-dc	edd55388ce	switch python load test to use load	2023-12-22 01:17:08 -05:00
dc-dc-dc	dc5abdc4c4	updated load api to take format as an option else infer from extension	2023-12-22 01:17:08 -05:00
dc-dc-dc	c81f5a5b94	added format mlx to metadata	2023-12-22 01:16:04 -05:00
dc-dc-dc	29e43170c4	added complex64 and python tests	2023-12-22 01:16:04 -05:00
Awni Hannun	3f82fe8888	fetch json	2023-12-22 01:16:04 -05:00
dc-dc-dc	c043d27682	more cleanup	2023-12-22 01:16:04 -05:00
dc-dc-dc	d3b1bea251	remove utils header	2023-12-22 01:16:04 -05:00
dc-dc-dc	472ce433f8	removed custom json parser for nlohmann	2023-12-22 01:16:04 -05:00
dc-dc-dc	99b9c1dac5	fix doc	2023-12-22 01:16:04 -05:00
dc-dc-dc	c88d3174aa	added save bindings and fixed header	2023-12-22 01:16:04 -05:00
dc-dc-dc	9a39254959	add python bindings for loading	2023-12-22 01:16:04 -05:00
dc-dc-dc	dcfa2700f6	use the correct type	2023-12-22 01:16:04 -05:00
dc-dc-dc	801c1ad3a4	ignore __metadata__	2023-12-22 01:16:04 -05:00
dc-dc-dc	f4d876d35f	load changes were not needed after all	2023-12-22 01:16:04 -05:00
dc-dc-dc	60132a16de	uint64_t instead of uint32_t	2023-12-22 01:16:04 -05:00
dc-dc-dc	c432de8013	removed debugging from load	2023-12-22 01:16:04 -05:00
dc-dc-dc	18a1c335d0	updated tests	2023-12-22 01:16:04 -05:00
dc-dc-dc	f09bcc7d50	saving works	2023-12-22 01:16:04 -05:00
dc-dc-dc	5aa0b1f632	expand dtype support	2023-12-22 01:16:04 -05:00
dc-dc-dc	ba869e5e71	remove safe file	2023-12-22 01:16:04 -05:00
dc-dc-dc	5b4155d4d0	more tests	2023-12-22 01:16:04 -05:00
dc-dc-dc	42baa095d1	its working	2023-12-22 01:16:04 -05:00
dc-dc-dc	9be3ea69ee	switch to unordered map	2023-12-22 01:16:04 -05:00
dc-dc-dc	fef579cec1	fixed array parsing	2023-12-22 01:16:04 -05:00
dc-dc-dc	91495382fd	more cleanup	2023-12-22 01:16:04 -05:00
dc-dc-dc	d0285db98c	parse the number from json	2023-12-22 01:16:04 -05:00
dc-dc-dc	be075a82bf	small cleanup	2023-12-22 01:16:04 -05:00
dc-dc-dc	bd422decc4	covering more cases	2023-12-22 01:16:04 -05:00
dc-dc-dc	87ec7b3cf9	initial commit	2023-12-22 01:16:04 -05:00
Angelos Katharopoulos	8385f93cea	Bumping the version (#256)	2023-12-21 18:33:14 -08:00
Awni Hannun	2118c3dbfa	fix (#255)	2023-12-21 18:18:41 -08:00
Awni Hannun	a002797d52	A temporary fix (#254)	2023-12-21 17:59:15 -08:00
Angelos Katharopoulos	1d053e0d1d	Fix the alibi test that was left unchanged (#252)	2023-12-21 14:59:25 -08:00
Hazem Essam	0aa65c7a6b	Added ALiBi implementation (#232)	2023-12-21 14:36:38 -08:00
Daniel Strobusch	794feb83df	support arange for bfloat16 (#245)	2023-12-21 14:33:43 -08:00
Angelos Katharopoulos	2c7df6795e	Make sure that arrays are freed when saving (#247)	2023-12-21 14:08:24 -08:00
Angelos Katharopoulos	b3916cbf2b	Improve names of quantization arguments (#235) ... * Change the default quantization group_size to 64 * Rename groups to group_size and width to bits	2023-12-20 16:53:53 -08:00
Angelos Katharopoulos	57fe918cf8	Adds C++ and nn quantization utilities (#230) ... * Add C++ de-/quantize ops * Add quantize functions to the docs and tests * Add a QuantizedLinear module	2023-12-20 14:17:38 -08:00
Justin Deschenaux	4912ff3ec2	Add Lion optimizer (#209) ... * Add Lion optimizer * Update acknowledgements also with past contributions	2023-12-20 13:54:58 -08:00
Awni Hannun	f40d17047d	Indexing bug (#233) ... * fix * test	2023-12-20 10:44:01 -08:00
Angelos Katharopoulos	2807c6aff0	Implements divide for integer types and adds floor_divide op (#228) ... * Add floor_divide * Add floor_divide to the tests * Add floor_divide to the docs	2023-12-19 20:12:19 -08:00
davidkoski	de892cb66c	fix for non-macos build issue on cblas.h (#227)	2023-12-19 17:01:59 -08:00
davidkoski	37024d899c	fixes for building with swiftpm (#225) ... - clbas is part of veclib (compile failure) - add SWIFTPM_BUNDLE #define to allow loading the metallib from a swiftpm resource bundle	2023-12-19 16:22:10 -08:00
Diogo	137f55bf28	fail early if readinto does not exist (#221)	2023-12-19 13:27:17 -08:00