Yu, G. X., Gao, Y., Golikov, P., & Pekhimenko, G. (2021). Habitat: A Runtime-Based Computational Performance Predictor for Deep Neural Network Training [Conference paper]. Proceedings of the 2021 USENIX Annual Technical Conference (USENIX ATC '21), 503–521