flax

38323b2b - Merge pull request #937 from levskaya:transform_improvements2