google / google/flax

Add IndRNN

Open
#4,133 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
7.3k
Forks
833
Avg merge
5h 11m
Merged PRs (30d)
5

Description

**Feature request:** Add `IndRNNCell` to flax's [recurrent cells](https://flax.readthedocs.io/en/latest/api_reference/flax.linen/layers.html#recurrent).

- [Independently Recurrent Neural Network (IndRNN): Building a Longer and Deeper RNN](https://openaccess.thecvf.com/content_cvpr_2018/html/Li_Independently_Recurrent_Neural_CVPR_2018_paper.html). CVPR 2018.

It currently has 1004 citations according to [Google Scholar](https://scholar.google.com/scholar?cites=5784188603612259222).

I can submit a PR for this.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.