flax

416b3e20 - Merge pull request #3425 from chiamp:rnn_warning