activity
20222026
most citedGalvatron: Efficient Transformer Training over Multiple GPUs Using Automatic Parallelism

57 citations · 161 across the 13 of their papers we have counts for

collaborators
Showing 2023Show all

6 papers · 1 filter

cs.IR2023

Model-enhanced Vector Index

Hailin Zhang, Yujing Wang, Qi Chen +16

Embedding-based retrieval methods construct vector indices to search for document representations that are most similar to the query representations. They are widely used in docume…

cs.LG2023★ 13 cited

Improving Automatic Parallel Training via Balanced Memory Workload Optimization

Yujie Wang, Youhe Jiang, Xupeng Miao +5

Transformer models have emerged as the leading approach for achieving state-of-the-art performance across various application domains, serving as the foundation for advanced large-…

cs.DC2023

OSDP: Optimal Sharded Data Parallel for Distributed Deep Learning

Youhe Jiang, Fangcheng Fu, Xupeng Miao +2

Large-scale deep learning models contribute to significant performance improvements on varieties of downstream tasks. Current data and model parallelism approaches utilize model re…

cs.CV2023

Accelerating Text-to-Image Editing via Cache-Enabled Sparse Diffusion Inference

Zihao Yu, Haoyang Li, Fangcheng Fu +2

Due to the recent success of diffusion models, text-to-image generation is becoming increasingly popular and achieves a wide range of applications. Among them, text-to-image editin…

cs.DC2023★ 38 cited

FlexMoE: Scaling Large-scale Sparse Pre-trained Model Training via Dynamic Device Placement

Xiaonan Nie, Xupeng Miao, Zilong Wang +5

With the increasing data volume, there is a trend of using large-scale pre-trained models to store the knowledge into an enormous number of model parameters. The training of these…

cs.LG2023

Angel-PTM: A Scalable and Economical Large-scale Pre-training System in Tencent

Xiaonan Nie, Yi Liu, Fangcheng Fu +5

Recent years have witnessed the unprecedented achievements of large-scale pre-trained models, especially the Transformer models. Many products and services in Tencent Inc., such as…