flax

587f6148 - Merge pull request #2897 from chiamp:frozen_dict_utility_fns