What are variable dicts, variable collections, and the mutable argument to Module.apply in Flax?

[avital]

Then link to it instead of the existing line TODO: Make “variable dict” design note, and link to it from here.

[marcvanzee]

Some possibly useful discussion on variable dicts: #2080