zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-12 07:14:34 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add optimizers (AdaMax, AdaDelta, RMSprop) and ordering optimizer classes (#142)

Browse Source 
* Add AdaMax, AdaDelta, RMSprop
This commit is contained in:
YUN, Junwoo
2023-12-17 13:43:15 +08:00
committed by GitHub
parent a67bbfe745
commit eebd7c275d
2 changed files with 219 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
docs/src/python/optimizers.rst
View File

@@ -38,4 +38,9 @@ model's parameters and the **optimizer state**.
OptimizerState
Optimizer
SGD
RMSprop
Adagrad
AdaDelta
Adam
AdamW
Adamax