flax

3da44bc1 - Merge pull request #3712 from carlosgmartin:conv_lstm_cell_docs