flax

6b70fe87 - Merge pull request #3670 from chiamp:vmap