zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-01 12:49:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Implement RNN, GRU, LSTM (#268)

Browse Source 
* 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>
This commit is contained in:
Justin Deschenaux
2024-03-12 05:14:44 +01:00
committed by GitHub
parent 0e95b64942
commit 8e5600022a
6 changed files with 361 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
docs/src/python/nn/layers.rst
View File

@@ -21,9 +21,11 @@ Layers
Embedding
GELU
GroupNorm
GRU
InstanceNorm
LayerNorm
Linear
LSTM
MaxPool1d
MaxPool2d
Mish
@@ -32,6 +34,7 @@ Layers
QuantizedLinear
RMSNorm
ReLU
RNN
RoPE
SELU
Sequential