Tianqi Chen
- Learning to Optimize Tensor Programs
2018/05/21 by Tianqi Chen, Lianmin Zheng, Chen, Tianqi +14 · 1 voice · 25 citations
Computer Science · Physics and Astronomy · #Parallel Computing and Optimization Techniques #Computational Physics and Python Applications #Model Reduction and Neural Networks
- Training Deep Nets with Sublinear Memory Cost
2016/04/21 by Tianqi Chen, Bing Xu, Chen, Tianqi +5 · 117 citations
Computer Science · #Advanced Neural Network Applications #Domain Adaptation and Few-Shot Learning #FOS: Computer and information sciences #Machine Learning (cs.LG) #Machine Learning and Data Classification
- MXNet: A Flexible and Efficient Machine Learning Library for Heterogeneous Distributed Systems
2015/12/03 by Tianqi Chen, Mu Li, Chen, Tianqi +17 · 130 citations
Engineering · Computer Science · #Advanced Memory and Neural Computing #Advanced Neural Network Applications #IoT and Edge/Fog Computing
- Stochastic Gradient Hamiltonian Monte Carlo
2014/02/17 by Tianqi Chen, Emily B. Fox, Chen, Tianqi +3 · 51 citations
Mathematics · Computer Science · #Markov Chains and Monte Carlo Methods #Gaussian Processes and Bayesian Inference #Stochastic Gradient Optimization Techniques
- TVM: An Automated End-to-End Optimizing Compiler for Deep Learning
2018/02/12 by Tianqi Chen, Thierry Moreau, Chen, Tianqi +22 · 1 voice · 14 citations
Computer Science · #Parallel Computing and Optimization Techniques #Advanced Neural Network Applications #Advanced Data Storage Technologies
- Atom: Low-bit Quantization for Efficient and Accurate LLM Serving
2023/10/29 by Yilong Zhao, Chien‐Yu Lin, Zhao, Yilong +17 · 40 citations
Computer Science · #FOS: Computer and information sciences #Machine Learning (cs.LG) #Multimodal Machine Learning Applications #Natural Language Processing Techniques #Topic Modeling
- FlashInfer: Efficient and Customizable Attention Engine for LLM Inference Serving
2025/01/02 by Zihao Ye, Ye, Zihao, Lequn Chen +19 · 68 citations
Computer Science · Decision Sciences · #Artificial Intelligence (cs.AI) #Data Quality and Management #Digital and Cyber Forensics #Distributed #FOS: Computer and information sciences #Machine Learning (cs.LG) #Parallel #Scientific Computing and Data Management #and Cluster Computing (cs.DC)
- Event Tensor: A Unified Abstraction for Compiling Dynamic Megakernel
2026/04/14 by Hongyi Jin, Bohan Hou, Guanjie Wang +18 · 4 voices
#cs.DC #cs.LG #cs.PL
- Towards Efficient Generative Large Language Model Serving: A Survey from Algorithms to Systems
2023/12/23 by Xupeng Miao, Gabriele Oliaro, Miao, Xupeng +11 · 14 citations
Computer Science · Materials Science · #Artificial Intelligence (cs.AI) #Distributed #FOS: Computer and information sciences #Machine Learning (cs.LG) #Machine Learning in Materials Science #Natural Language Processing Techniques #Parallel #Performance (cs.PF) #Topic Modeling #and Cluster Computing (cs.DC)
- Tensor Program Optimization with Probabilistic Programs
2022/05/26 by Junru Shao, Xiyou Zhou, Shao, Junru +17 · 10 citations
Computer Science · Mathematics · #Parallel Computing and Optimization Techniques #Computational Physics and Python Applications #Tensor decomposition and applications
- SparseTIR: Composable Abstractions for Sparse Compilation in Deep Learning
2022/07/11 by Zihao Ye, Ruihang Lai, Ye, Zihao +7 · 6 citations
Computer Science · Mathematics · #Advanced Neural Network Applications #Artificial Intelligence (cs.AI) #FOS: Computer and information sciences #Machine Learning (cs.LG) #Parallel Computing and Optimization Techniques #Programming Languages (cs.PL) #Tensor decomposition and applications
- XGrammar: Flexible and Efficient Structured Generation Engine for Large Language Models
2024/11/22 by Yixin Dong, Charlie F. Ruan, Dong, Yixin +11 · 13 citations
Computer Science · #Artificial Intelligence (cs.AI) #Computation and Language (cs.CL) #FOS: Computer and information sciences #Natural Language Processing Techniques #Programming Languages (cs.PL) #Topic Modeling
- APE: Faster and Longer Context-Augmented Generation via Adaptive Parallel Encoding
2025/02/08 by Xinyu Yang, Yang, Xinyu, Tianqi Chen +3 · 8 citations
Computer Science · #Advanced Image and Video Retrieval Techniques #Advanced Neural Network Applications #Multimodal Machine Learning Applications
- Cancer-Myth: Evaluating Large Language Models on Patient Questions with False Presuppositions
2025/04/15 by Wang Bill Zhu, Wang Zhu, Tianqi Chen +18 · 1 voice · 5 citations
Computer Science · Medicine · #Artificial Intelligence in Healthcare and Education #Machine Learning in Healthcare #Topic Modeling #cs.CL #cs.CY
- 2.5D Multi-view Averaging Diffusion Model for 3D Medical Image Translation: Application to Low-count PET Reconstruction with CT-less Attenuation Correction
2024/06/12 by Tianqi Chen, Chen, Tianqi, Jun Hou +19 · 4 citations
Computer Science · Medicine · #Artificial Intelligence (cs.AI) #Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #FOS: Electrical engineering #Image and Video Processing (eess.IV) #Medical Image Segmentation Techniques #Medical Imaging Techniques and Applications #Radiomics and Machine Learning in Medical Imaging #electronic engineering #information engineering
- SOL-ExecBench: Speed-of-Light Benchmarking for Real-World GPU Kernels Against Hardware Limits
2026/03/19 by Edward Lin, Sahil Modi, Siva Kumar Sastry Hari +30 · 1 voice · 1 citation
Computer Science · #cs.LG #cs.AI
- Tilus: A Tile-Level GPGPU Programming Language for Low-Precision Computation
2025/04/17 by Yaoyao Ding, Bohan Hou, Ding, Yaoyao +13 · 4 citations
Computer Science · #Parallel Computing and Optimization Techniques #Advanced Neural Network Applications #Natural Language Processing Techniques
- RAILS: A Robust Adversarial Immune-inspired Learning System
2021/06/27 by Ren Wang, Wang, Ren, Tianqi Chen +11 · 1 citation
Biochemistry, Genetics and Molecular Biology · Computer Science · #Adversarial Robustness in Machine Learning #Bacillus and Francisella bacterial research #Cryptography and Security (cs.CR) #FOS: Computer and information sciences #Machine Learning (cs.LG) #Neural and Evolutionary Computing (cs.NE)
- A Hardware-Software Blueprint for Flexible Deep Learning Specialization
2018/07/11 by Thierry Moreau, Moreau, Thierry, Tianqi Chen +19 · 1 voice
Computer Science · Mathematics · #Distributed #FOS: Computer and information sciences #Machine Learning (cs.LG) #Machine Learning (stat.ML) #Parallel #and Cluster Computing (cs.DC) #cs.DC #cs.LG #stat.ML
- Bayesian network mediation analysis with application to brain functional connectome
2022/01/27 by Yize Zhao, Zhao, Yize, Tianqi Chen +9 · 1 citation
Neuroscience · Psychology · Physics and Astronomy · #Functional Brain Connectivity Studies #Mental Health Research Topics #Complex Network Analysis Techniques
- WebLLM: A High-Performance In-Browser LLM Inference Engine
2024/12/20 by Ruan, Charlie F., Qin, Yucheng, Parthasarathy, Akaash R. +13 · 2 citations
Computer Science · #Artificial Intelligence (cs.AI) #FOS: Computer and information sciences #Machine Learning (cs.LG) #Natural Language Processing Techniques #Speech Recognition and Synthesis
- Cascaded Multi-path Shortcut Diffusion Model for Medical Image Translation
2024/04/06 by Yinchi Zhou, Zhou, Yinchi, Tianqi Chen +17 · 1 citation
Medicine · #Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #FOS: Electrical engineering #Image and Video Processing (eess.IV) #Radiomics and Machine Learning in Medical Imaging #electronic engineering #information engineering
- Robust Simulations of Many-Body Symmetry-Protected Topological Phase Transitions on a Quantum Processor
2025/03/11 by Ruizhe Shen, Shen, Ruizhe, Tianqi Chen +7 · 2 citations
Computer Science · Physics and Astronomy · #FOS: Physical sciences #Quantum Computing Algorithms and Architecture #Quantum Physics (quant-ph) #Quantum and electron transport phenomena #Quantum many-body systems #Strongly Correlated Electrons (cond-mat.str-el)
- MPK: A Compiler and Runtime for Mega-Kernelizing Tensor Programs
2025/12/22 by Xinhao Cheng, Zhihao Zhang, Cheng, Xinhao +37 · 2 citations
Computer Science · #Advanced Neural Network Applications #Parallel Computing and Optimization Techniques #Graph Theory and Algorithms