mlx/python/mlx/optimizers
Chime Ogbuji f5a1582fe8
Add minimum for cosine decay function (#859)
* Add minimum for cosine decay function

* Update python/mlx/optimizers/schedulers.py

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

---------

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
2024-03-21 07:33:29 -07:00
..
__init__.py Support LR schedulers (#334) 2024-02-15 11:26:20 -08:00
optimizers.py Some fixes in cache / thread safety (#777) 2024-03-05 13:30:50 -08:00
schedulers.py Add minimum for cosine decay function (#859) 2024-03-21 07:33:29 -07:00