flax

Fix bug that assumed frozen-dict keys were strings.

#3692

Merged