Follow
Ajay Jain
Title
Cited by
Cited by
Year
Denoising diffusion probabilistic models
J Ho, A Jain, P Abbeel
Advances in neural information processing systems 33, 6840-6851, 2020
132572020
Dreamfusion: Text-to-3d using 2d diffusion
B Poole, A Jain, JT Barron, B Mildenhall
arXiv preprint arXiv:2209.14988, 2022
15542022
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
4762021
Putting NeRF on a Diet: Semantically Consistent Few-Shot View Synthesis
A Jain, M Tancik, P Abbeel
International Conference on Computer Vision, 2021
4382021
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
1822019
Contrastive code representation learning
P Jain, A Jain, T Zhang, P Abbeel, JE Gonzalez, I Stoica
arXiv preprint arXiv:2007.04973, 2020
1422020
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
872018
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
802019
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
552020
Locally Masked Convolution for Autoregressive Models
A Jain, P Abbeel, D Pathak
Conference on Uncertainty in Artificial Intelligence (UAI), 2020
312020
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
242024
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
182019
Revec: Program Rejuvenation through Revectorization
C Mendis*, A Jain*, P Jain, S Amarasinghe
Proceedings of the 28th International Conference on Compiler Construction, 2019
172019
An analysis of the delayed gradients problem in asynchronous sgd
A Srinivasan, A Jain, P Barekatain
102018
Autonomy for surface ship interception
C Mirabito, DN Subramani, T Lolla, PJ Haley, A Jain, PFJ Lermusiaux, ...
OCEANS 2017-Aberdeen, 1-10, 2017
82017
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
42019
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
32022
Learning Automatic Schedulers with Projective Reparameterization
A Jain, S Amarasinghe
ISCA Workshop on Machine Learning for Systems, 2019
22019
The system can't perform the operation now. Try again later.
Articles 1–18