flax
renamed channel_axes to feature_axes in InstanceNorm
#3667
Merged

Loading