flax

8a9fbc9c - Merge pull request #2701 from cgarciae:cleanup-dense-general