Representing Long-Range Context for Graph Neural Networks with Global Attention Z Wu, P Jain, M Wright, A Mirhoseini, JE Gonzalez, I Stoica NeurIPS 2021, 2021 | 191 | 2021 |
Checkmate: Breaking the Memory Wall with Optimal Tensor Rematerialization P Jain, A Jain, A Nrusimha, A Gholami, P Abbeel, K Keutzer, I Stoica, ... Proceedings of Machine Learning and Systems 2020, 497--511, 2020 | 154 | 2020 |
Contrastive Code Representation Learning P Jain, A Jain, T Zhang, P Abbeel, JE Gonzalez, I Stoica EMNLP 2021, 2021 | 129 | 2021 |
Data synthesis for autonomous control systems FN Iandola, DB MacMillen, A Shen, HS Sidhu, PJ Jain US Patent 10,678,244, 2020 | 85 | 2020 |
Dynamic Space-Time Scheduling for GPU Inference P Jain, X Mo, A Jain, H Subbaraj, R Durrani, A Tumanov, J Gonzalez, ... LearningSys Workshop at Neural Information Processing Systems 2018, 2018 | 77 | 2018 |
Multi-channel sensor simulation for autonomous control systems FN Iandola, DB MacMillen, A Shen, HS Sidhu, DP Tomasello, RN Phadte, ... US Patent 11,157,014, 2021 | 46 | 2021 |
POET: Training neural networks on tiny devices with integrated rematerialization and paging SG Patil, P Jain, P Dutta, I Stoica, J Gonzalez International Conference on Machine Learning, 17573-17583, 2022 | 29 | 2022 |
Accelerating Quadratic Optimization with Reinforcement Learning J Ichnowski, P Jain, B Stellato, G Banjac, M Luo, F Borrelli, JE Gonzalez, ... NeurIPS 2021, 2021 | 24 | 2021 |
Skyplane: Optimizing transfer cost and throughput using {Cloud-Aware} overlays P Jain, S Kumar, S Wooders, SG Patil, JE Gonzalez, I Stoica 20th USENIX Symposium on Networked Systems Design and Implementation (NSDI …, 2023 | 20 | 2023 |
Revec: program rejuvenation through revectorization C Mendis, A Jain, P Jain, S Amarasinghe Proceedings of the 28th International Conference on Compiler Construction, 29-41, 2019 | 17 | 2019 |
The OoO VLIW JIT compiler for GPU inference P Jain, X Mo, A Jain, A Tumanov, JE Gonzalez, I Stoica arXiv preprint arXiv:1901.10008, 2019 | 17 | 2019 |
Analyzing and comparing lakehouse storage systems P Jain, P Kraft, C Power, T Das, I Stoica, M Zaharia Proc. Conf. Innov. Data Syst. Res, 2023 | 11 | 2023 |
Grounded Graph Decoding improves Compositional Generalization in Question Answering Y Gai, P Jain, W Zhang, JE Gonzalez, I Stoica, D Song Findings of EMNLP 2021, 2021 | 8 | 2021 |
DSCnet: Replicating lidar point clouds with deep sensor cloning P Tomasello, S Sidhu, A Shen, MW Moskewicz, N Redmon, G Joshi, ... Proceedings of the IEEE Conference on Computer Vision and Pattern …, 2019 | 8 | 2019 |
Spotting Suspicious Reviews via (Quasi-) clique Extraction P Jain, ST Chen, M Azimpourkivi, DH Chau, B Carbunar IEEE Security and Privacy 2015 extended abstract, 2015 | 4 | 2015 |
Systems and methods for training machine models with augmented data MJ Cooper, PJ Jain, HS Sidhu US Patent 11,205,093, 2021 | 1 | 2021 |
Systems and methods for training machine models with augmented data MJ Cooper, PJ Jain, HS Sidhu US Patent 11,893,774, 2024 | | 2024 |
Optimizing neural network structures for embedded systems HS Sidhu, PJ Jain, DP Tomasello, FN Iandola US Patent 11,636,333, 2023 | | 2023 |
A Framework for Synthesizing Low-Power Approximate Inference Accelerators P Jain, S Huda, M Maas, JE Gonzalez, I Stoica, A Mirhoseini DATE 2022, 2022 | | 2022 |
Synthesizing Low-Power Approximate Hardware with Large-Scale Search P Jain, S Huda, M Maas, J Gonzalez, I Stoica, A Mirhoseini MLArchSys at the International Symposium on Computer Architecture 2021, 2021 | | 2021 |