flax

19c4e273 - Merge pull request #1535 from jheek:remove-implicit-rank-promotion