flax

f5b37967 - add rnnbase refactor flip