mlx/python
Justin Deschenaux 8e5600022a
Implement RNN, GRU, LSTM (#268)
* RNN base implementation

* Address comments+format

* nits in docs

* add tests for prb

* fix test

* add a couple tests

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2024-03-11 21:14:44 -07:00
..
mlx Implement RNN, GRU, LSTM (#268) 2024-03-11 21:14:44 -07:00
src Version bump and os error (#807) 2024-03-07 13:57:58 -08:00
tests Implement RNN, GRU, LSTM (#268) 2024-03-11 21:14:44 -07:00