mlx/docs/src
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