Anchen f6feb61f92 feat: add support for saving safetensors in the save_weights (#497) ... * feat: add save safetensors support in module save_weights * chore: checking missing changes * Update python/mlx/nn/layers/base.py Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * chore: update docstring for load_weights --------- Co-authored-by: Awni Hannun <awni.hannun@gmail.com>		2024-01-19 06:19:33 -08:00
..
nn	feat: add support for saving safetensors in the save_weights (#497)	2024-01-19 06:19:33 -08:00
_reprlib_fix.py	copyright + ack	2023-11-30 11:12:53 -08:00
extension.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
optimizers.py	Spelling (#342)	2024-01-01 21:08:17 -08:00
utils.py	Minor refactor for tree_map and tree_unflatten (#311)	2023-12-28 20:55:10 -08:00