Denoising diffusion probabilistic models J Ho, A Jain, P Abbeel Advances in neural information processing systems 33, 6840-6851, 2020 | 11732 | 2020 |
Dreamfusion: Text-to-3d using 2d diffusion B Poole, A Jain, JT Barron, B Mildenhall arXiv preprint arXiv:2209.14988, 2022 | 1415 | 2022 |
Zero-Shot Text-Guided Object Generation with Dream Fields A Jain, B Mildenhall, JT Barron, P Abbeel, B Poole Conference on Computer Vision and Pattern Recognition (CVPR 2022), 2021 | 447 | 2021 |
Putting NeRF on a Diet: Semantically Consistent Few-Shot View Synthesis A Jain, M Tancik, P Abbeel International Conference on Computer Vision, 2021 | 409 | 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 the 3rd Conference on Machine Learning and Systems (MLSys), 2019 | 170 | 2019 |
Contrastive code representation learning P Jain, A Jain, T Zhang, P Abbeel, JE Gonzalez, I Stoica arXiv preprint arXiv:2007.04973, 2020 | 135 | 2020 |
Dynamic Space-Time Scheduling for GPU Inference P Jain, X Mo, A Jain, H Subbaraj, RS Durrani, A Tumanov, J Gonzalez, ... Advances in Neural Information Processing Systems workshop on Systems for …, 2018 | 81 | 2018 |
Discrete Residual Flow for Probabilistic Pedestrian Behavior Prediction A Jain, S Casas Romero, R Liao, Y Xiong, S Feng, S Segal, R Urtasun Proceedings of the 3rd Conference on Robot Learning, 2019 | 80 | 2019 |
Sparse graphical memory for robust planning S Emmons, A Jain, M Laskin, T Kurutach, P Abbeel, D Pathak Advances in neural information processing systems 33, 5251-5262, 2020 | 55 | 2020 |
Locally Masked Convolution for Autoregressive Models A Jain, P Abbeel, D Pathak Conference on Uncertainty in Artificial Intelligence (UAI), 2020 | 31 | 2020 |
Video prediction models as rewards for reinforcement learning A Escontrela, A Adeniji, W Yan, A Jain, XB Peng, K Goldberg, Y Lee, ... Advances in Neural Information Processing Systems 36, 2024 | 20 | 2024 |
Revec: Program Rejuvenation through Revectorization C Mendis*, A Jain*, P Jain, S Amarasinghe Proceedings of the 28th International Conference on Compiler Construction, 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 |
An analysis of the delayed gradients problem in asynchronous sgd A Srinivasan, A Jain, P Barekatain | 10 | 2018 |
Autonomy for surface ship interception C Mirabito, DN Subramani, T Lolla, PJ Haley, A Jain, PFJ Lermusiaux, ... OCEANS 2017-Aberdeen, 1-10, 2017 | 8 | 2017 |
Using effective dimension to analyze feature transformations in deep neural networks K Ravichandran, A Jain, A Rakhlin ICML Workshop on Identifying and Understanding Deep Learning Phenomena, 2019 | 4 | 2019 |
Journey to the BAOAB-limit: finding effective MCMC samplers for score-based models A Jain, B Poole NeurIPS 2022 Workshop on Score-Based Methods, 2022 | 3 | 2022 |
Learning Automatic Schedulers with Projective Reparameterization A Jain, S Amarasinghe ISCA Workshop on Machine Learning for Systems, 2019 | 2 | 2019 |