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