flax

cba70922 - Merge pull request #1581 from jheek:map-variables-tranfsorm