Revec: program rejuvenation through revectorization. - Charith Mendis; Ajay Jain; Paras Jain 001; Saman P. Amarasinghe (CC, 2019)

DSCnet: Replicating Lidar Point Clouds With Deep Sensor Cloning. - Paden Tomasello; Sammy Sidhu; Anting Shen; Matthew W. Moskewicz; Nobie Redmon; Gayatri Joshi; Romi Phadte; Paras Jain 001; Forrest N. Iandola (CVPR Workshops, 2019)

Dynamic Space-Time Scheduling for GPU Inference. - Paras Jain 001; Xiangxi Mo; Ajay Jain; Harikaran Subbaraj; Rehan Sohail Durrani; Alexey Tumanov; Joseph Gonzalez 001; Ion Stoica (CoRR, 2019)

The OoO VLIW JIT Compiler for GPU Inference. - Paras Jain 001; Xiangxi Mo; Ajay Jain; Alexey Tumanov; Joseph E. Gonzalez; Ion Stoica (CoRR, 2019)

Checkmate: Breaking the Memory Wall with Optimal Tensor Rematerialization. - Paras Jain 001; Ajay Jain; Aniruddha Nrusimha; Amir Gholami; Pieter Abbeel; Kurt Keutzer; Ion Stoica; Joseph Gonzalez 001 (MLSys, 2020)

Contrastive Code Representation Learning. - Paras Jain 001; Ajay Jain; Tianjun Zhang; Pieter Abbeel; Joseph Gonzalez 001; Ion Stoica (EMNLP (1), 2021)

Grounded Graph Decoding improves Compositional Generalization in Question Answering. - Yu Gai; Paras Jain 001; Wendi Zhang; Joseph Gonzalez 001; Dawn Song; Ion Stoica (EMNLP (Findings), 2021)

Accelerating Quadratic Optimization with Reinforcement Learning. - Jeffrey Ichnowski; Paras Jain 001; Bartolomeo Stellato; Goran Banjac; Michael Luo; Francesco Borrelli; Joseph E. Gonzalez; Ion Stoica; Ken Goldberg (NeurIPS, 2021)

Representing Long-Range Context for Graph Neural Networks with Global Attention. - Zhanghao Wu; Paras Jain 001; Matthew A. Wright; Azalia Mirhoseini; Joseph E. Gonzalez; Ion Stoica (NeurIPS, 2021)

Learning to Design Accurate Deep Learning Accelerators with Inaccurate Multipliers. - Paras Jain 001; Safeen Huda; Martin Maas 001; Joseph E. Gonzalez; Ion Stoica; Azalia Mirhoseini (DATE, 2022)

POET: Training Neural Networks on Tiny Devices with Integrated Rematerialization and Paging. - Shishir G. Patil; Paras Jain 001; Prabal Dutta; Ion Stoica; Joseph Gonzalez 001 (ICML, 2022)

Analyzing and Comparing Lakehouse Storage Systems. - Paras Jain 001; Peter Kraft; Conor Power; Tathagata Das; Ion Stoica; Matei Zaharia (CIDR, 2023)

Skyplane: Optimizing Transfer Cost and Throughput Using Cloud-Aware Overlays. - Paras Jain 001; Sam Kumar; Sarah Wooders; Shishir G. Patil; Joseph E. Gonzalez; Ion Stoica (NSDI, 2023)

Towards Robust and Scalable Large Language Models - Paras Jain 001 (2023)

Cloudcast: High-Throughput, Cost-Aware Overlay Multicast in the Cloud. - Sarah Wooders; Shu Liu; Paras Jain 001; Xiangxi Mo; Joseph E. Gonzalez; Vincent Liu 001; Ion Stoica (NSDI, 2024)