Z. Yao, A. Gholami, S. Shen, M. Mustafa, K. Keutzer, and M. Mahoney, “ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning”, AAAI, vol. 35, no. 12, pp. 10665–10673, May 2021.