Giters

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

Feature request: Add ConvGRUCell

carlosgmartin opened this issue · comments

commented

Add the Convolutional GRU to flax.linen. This is the GRU counterpart to flax.linen.ConvLSTMCell.