Hazem Essam 37fc9db82c Added Adafactor (#415) ...

* Added adafactor

* Added Adafactor and ran pre-commit

* modified operations

* Added docstrings

* Switched two ops to fix a bug

* added underscore for internal functions and removed the plus sign in the last return statment

* Removed parameter rms from the optimizer state because its not needed

* Added simple MNIST test for Adafactor and temporary training log

* remove test files

* nits in docs

* comment nit

---------

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

2024-01-23 15:11:27 -08:00

..

_static

Doc theme (#5)

2023-12-05 12:08:05 -08:00

_templates

Module checks the weight on load_weights (#337)

2024-01-02 18:55:42 -08:00

cpp

jagrit's commit files

2023-11-29 10:52:08 -08:00

dev

Spelling (#342)

2024-01-01 21:08:17 -08:00

examples

Update mlx-example link for llms llama in llama-inference.rst (#405)

2024-01-08 16:29:53 -08:00

python

Added Adafactor (#415)

2024-01-23 15:11:27 -08:00

usage

usage doc for function transformations (#481)

2024-01-17 17:10:53 -08:00

conf.py

version bump (#490)

2024-01-18 12:00:24 -08:00

index.rst

usage doc for function transformations (#481)

2024-01-17 17:10:53 -08:00

install.rst

docs (#444)

2024-01-12 13:34:16 -08:00