Single GPU
Single-GPU computing remains a crucial area of research, focusing on optimizing the performance and energy efficiency of various machine learning tasks, including large language model (LLM) inference, image generation, and other computationally intensive algorithms. Current research emphasizes efficient memory management, novel attention mechanisms (like linear attention), and optimized kernel designs to maximize throughput and minimize latency, often targeting specific model architectures like Transformers and diffusion models. These advancements are significant because they enable cost-effective deployment of powerful AI models on readily available hardware, broadening access to advanced computational capabilities and accelerating progress across diverse scientific and industrial applications.
Papers
Forecasting GPU Performance for Deep Learning Training and Inference
Seonho Lee, Amar Phanishayee, Divya Mahajan
LiNR: Model Based Neural Retrieval on GPUs at LinkedIn
Fedor Borisyuk, Qingquan Song, Mingzhou Zhou, Ganesh Parameswaran, Madhu Arun, Siva Popuri, Tugrul Bingol, Zhuotao Pei, Kuang-Hsuan Lee, Lu Zheng, Qizhan Shao, Ali Naqvi, Sen Zhou, Aman Gupta
MEMO: Fine-grained Tensor Management For Ultra-long Context LLM Training
Pinxue Zhao, Hailin Zhang, Fangcheng Fu, Xiaonan Nie, Qibin Liu, Fang Yang, Yuanbo Peng, Dian Jiao, Shuaipeng Li, Jinbao Xue, Yangyu Tao, Bin Cui
Characterizing and Understanding HGNN Training on GPUs
Dengke Han, Mingyu Yan, Xiaochun Ye, Dongrui Fan