More numerically stable batch norm. Fixes #182 #183

Closed david-berthelot closed 3 years ago