flax

6759522a - Fix NamedTuples being converted to tuples in Module

Fix NamedTuples being converted to tuples in Module NamedTuples are constructed differently so they have to be special cased by the freezing logic in Module (*args instead of Iterable)