Robust Training of Neural Networks Using Scale Invariant Architectures. - Li, Zhiyuan; Bhojanapalli, Srinadh; Zaheer, Manzil; Reddi, Sashank J.; Kumar, Sanjiv (2022)