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