flax

f19a720a - Merge pull request #1237 from jheek:fix-dense-general