Bai, Junjie
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
2019/12/03 by Adam Paszke, Paszke, Adam, Sam Gross +41 · 1 voice · 4266 citations
Computer Science · Mathematics · #Computational Physics and Python Applications #Machine Learning and Data Classification #Parallel Computing and Optimization Techniques #cs.LG #cs.MS #stat.ML
- DistriFusion: Distributed Parallel Inference for High-Resolution Diffusion Models
2024/02/29 by Muyang Li, Li, Muyang, Tianle Cai +17 · 35 citations
Computer Science · Mathematics · Medicine · #Advanced Mathematical Modeling in Engineering #Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #Medical Imaging Techniques and Applications #Statistical Methods and Inference
- Cosmos-Reason1: From Physical Common Sense To Embodied Reasoning
2025/03/18 by NVIDIA, :, Azzolini, Alisson +51 · 60 citations
#Artificial Intelligence (cs.AI) #Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #Machine Learning (cs.LG) #Robotics (cs.RO)
- OmniMotionGPT: Animal Motion Generation with Limited Data
2023/11/30 by Zhangsihao Yang, Mingyuan Zhou, Yang, Zhangsihao +15 · 9 citations
Computer Science · Engineering · #Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #Generative Adversarial Networks and Image Synthesis #Human Motion and Animation #Human Pose and Action Recognition
- Parameter-Efficient Sparsity for Large Language Models Fine-Tuning
2022/05/23 by Yuchao Li, Li, Yuchao, Fuli Luo +11 · 5 citations
Computer Science · #Topic Modeling #Natural Language Processing Techniques #Multimodal Machine Learning Applications
- DISC: A Dynamic Shape Compiler for Machine Learning Workloads
2021/03/09 by Kai Zhu, Zhu, Kai, Wenyi Zhao +18 · 3 citations
Computer Science · #Advanced Image and Video Retrieval Techniques #Advanced Neural Network Applications #Distributed #FOS: Computer and information sciences #Parallel #Parallel Computing and Optimization Techniques #and Cluster Computing (cs.DC)
- World Simulation with Video Foundation Models for Physical AI
2025/10/28 by Nvidia Nvidia, NVIDIA, Arslan Ali +170 · 18 citations
Computer Science · #Multimodal Machine Learning Applications #Generative Adversarial Networks and Image Synthesis #Reinforcement Learning in Robotics
- Alpamayo-R1: Bridging Reasoning and Action Prediction for Generalizable Autonomous Driving in the Long Tail
2025/10/30 by Nvidia Nvidia, NVIDIA, Yan Wang +78 · 15 citations
Computer Science · Engineering · #Adversarial Robustness in Machine Learning #Artificial Intelligence (cs.AI) #Autonomous Vehicle Technology and Safety #FOS: Computer and information sciences #Machine Learning (cs.LG) #Reinforcement Learning in Robotics #Robotics (cs.RO)
- DeepCenterline: a Multi-task Fully Convolutional Network for Centerline Extraction
2019/03/25 by Guo, Zhihui, Bai, Junjie, Lu, Yi +5 · 1 citation
#Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences
- Attention-driven Tree-structured Convolutional LSTM for High Dimensional Data Understanding
2019/01/29 by Kong, Bin, Wang, Xin, Bai, Junjie +7 · 1 citation
#Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences