flax

a0a57ebf - Merge pull request #2767 from cgarciae:rename-conv-lstm