google / flax

Flax is a neural network library for JAX that is designed for flexibility.

Home Page:https://flax.readthedocs.io

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Add IndRNN

carlosgmartin opened this issue · comments

Feature request: Add IndRNNCell to flax's recurrent cells.

It currently has 1004 citations according to Google Scholar.

I can submit a PR for this.

One big factor against adding IndRNN to Flax is that neither Pytorch nor Keras provides this layer.