flax

27c37ed1 - Merge pull request #2872 from chiamp:frozen_dict_feature_flag