mlx/python/mlx
Awni Hannun 420ff2f331
Add back compiled function signatures and docstrings (#749)
* try to add back compiled function signatures and docstrings

* add indentation to docstring
2024-02-27 13:18:59 -08:00
..
nn Add back compiled function signatures and docstrings (#749) 2024-02-27 13:18:59 -08:00
optimizers Add linear warmup and schedule joining for use with existing schedules (#721) 2024-02-26 07:28:48 -08:00
_reprlib_fix.py copyright + ack 2023-11-30 11:12:53 -08:00
extension.py Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00
py.typed Add py.typed to support PEP-561 (type-hinting) for mlx (#588) 2024-01-31 12:05:42 -08:00
utils.py Adds device context manager (#679) 2024-02-14 14:14:58 -08:00