-
Habitat: A Runtime-Based Computational Performance Predictor for Deep Neural Network Training
Geoffrey X. Yu, Yubo Gao, Pavel Golikov, Gennady Pekhimenko
Proceedings of the 2021 USENIX Annual Technical Conference (USENIX ATC'21), 2021. [Link]
-
Proteus: Preserving Model Confidentiality during Graph Optimizations
Yubo Gao, Maryam Haghifam, Christina Giannoula, Renbo Tu, Gennady Pekhimenko, Nandita Vijaykumar
Proceedings of Machine Learning and Systems 6 (MLSys 2024), 2024. [Link] [Talk]
-
APPL: A Prompt Programming Language for Harmonious Integration of Programs and Large Language Model Prompts
Honghua Dong, Qidong Su, Yubo Gao, Zhaoyu Li, Yangjun Ruan, Gennady Pekhimenko, Chris J. Maddison, Xujie Si
arXiv preprint arXiv:2406.13161, 2024. [Link]