The Limit of the Batch Size - Yang You; Yuhui Wang; Huan Zhang; Zhao Zhang; James Demmel; Cho-Jui Hsieh (2020)