flax
fix Variable pytree flatten
#4962
Merged

Loading