← Search

Junjie Bai

4 accepted papers

2024

DistriFusion: Distributed Parallel Inference for High-Resolution Diffusion Models

CVPR 2024highlight

Diffusion models have achieved great success in synthesizing high-quality images. However generating high-resolution images with diffusion models is still challenging due to the enormous computational costs resulting in a prohibitive latency for interactive applications. In this paper we propose Dis…

2024

OmniMotionGPT: Animal Motion Generation with Limited Data

CVPR 2024poster

Our paper aims to generate diverse and realistic animal motion sequences from textual descriptions without a large-scale animal text-motion dataset. While the task of text-driven human motion synthesis is already extensively studied and benchmarked it remains challenging to transfer this success to…

Cited by 7SourcePDFScholar
2022

Parameter-Efficient Sparsity for Large Language Models Fine-Tuning

IJCAI 2022poster

With the dramatically increased number of parameters in language models, sparsity methods have received ever-increasing research focus to compress and accelerate the models. While most research focuses on how to accurately retain appropriate weights while maintaining the performance of the compresse…

2019

PyTorch: An Imperative Style, High-Performance Deep Learning Library

NeurIPS 2019poster

Deep learning frameworks have often focused on either usability or speed, but not both. PyTorch is a machine learning library that shows that these two goals are in fact compatible: it was designed from first principles to support an imperative and Pythonic programming style that supports code as a…