mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-26 18:51:14 +08:00
818cda16bc
* Add a few LR schedulers * Move parents's constructor call to the top * Fix docstring * refactor optimizers into two files * add docs * nit * Fix Callable type annotation for python 3.8 --------- Co-authored-by: Awni Hannun <awni@apple.com> Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
5 lines
117 B
Python
5 lines
117 B
Python
# Copyright © 2023-2024 Apple Inc.
|
|
|
|
from mlx.optimizers.optimizers import *
|
|
from mlx.optimizers.schedulers import *
|