I recently came across switchable normalization in Andrew Ng’s ‘The Batch.’ Is this a layer that can be plugged into an existing model for normalization purposes? If so, is there a pytorch implementation of it?