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 | 227 | 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 | 164 | 2020 |
Contrastive Code Representation Learning P Jain, A Jain, T Zhang, P Abbeel, JE Gonzalez, I Stoica EMNLP 2021, 2021 | 133 | 2021 |
Data synthesis for autonomous control systems FN Iandola, DB MacMillen, A Shen, HS Sidhu, PJ Jain US Patent 10,678,244, 2020 | 89 | 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 | 79 | 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 | 47 | 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 | 38 | 2022 |
Accelerating Quadratic Optimization with Reinforcement Learning J Ichnowski, P Jain, B Stellato, G Banjac, M Luo, F Borrelli, JE Gonzalez, ... NeurIPS 2021, 2021 | 28 | 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 | 25 | 2023 |
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 |
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 | 16 | 2019 |
Analyzing and Comparing Lakehouse Storage Systems. P Jain, P Kraft, C Power, T Das, I Stoica, M Zaharia CIDR, 2023 | 14 | 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 | 9 | 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 |