Fix mutable_filter in map_variables
Fix mutable_filter in map_variables

The introduction of mutable filters wrongly disabled mutability on all collections for map_variables. Even those which aren't touched by the mapping.

PiperOrigin-RevId: 432005975

google/flax,google/flax