flax

cb907278 - Merge pull request #456 from jheek:fix-frozen-dict