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
..
mlx	Support Chaining for some of functionalities of nn.Module (#885) (#897)	2024-03-27 19:58:29 -07:00
src	Fix unsqueeze with None (#899)	2024-03-26 13:59:44 -07:00
tests	Support Chaining for some of functionalities of nn.Module (#885) (#897)	2024-03-27 19:58:29 -07:00