AmirHossein_Razlighi d611251502 Support Chaining for some of functionalities of nn.Module (#885) (#897) ... * add chaining support for some of the functionalities of "nn.Module" * reformat * change the return types * remove return types * add return type with forward referencing * add tests for chaining * add name to contributors * Update python/mlx/nn/layers/base.py Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * Update python/mlx/nn/layers/base.py Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * update docstring * update docstrings --------- Co-authored-by: Awni Hannun <awni.hannun@gmail.com>		2024-03-27 19:58:29 -07:00
..
nn	Support Chaining for some of functionalities of nn.Module (#885) (#897)	2024-03-27 19:58:29 -07:00
optimizers	Add minimum for cosine decay function (#859)	2024-03-21 07:33:29 -07:00
_os_warning.py	Version bump and os error (#807)	2024-03-07 13:57:58 -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