Online flow size prediction for improved network routing. - Pascal Poupart; Zhitang Chen; Priyank Jaini; Fred Fung; Hengky Susanto; Yanhui Geng; Li Chen 008; Kai Chen 005; Hao Jin (ICNP, 2016)

Online Algorithms for Sum-Product Networks with Continuous Variables. - Priyank Jaini; Abdullah Rashwan; Han Zhao 002; Yue Liu; Ershad Banijamali; Zhitang Chen; Pascal Poupart (Probabilistic Graphical Models, 2016)

Online and Distributed learning of Gaussian mixture models by Bayesian Moment Matching. - Priyank Jaini; Pascal Poupart (CoRR, 2016)

Online Bayesian Transfer Learning for Sequential Data Modeling. - Priyank Jaini; Zhitang Chen; Pablo Carbajal; Edith Law; Laura Middleton; Kayla Regan; Mike Schaekermann; George Trimponias; James Tung; Pascal Poupart (ICLR (Poster), 2017)

Accuracy Maximization Analysis for Sensory-Perceptual Tasks: Computational Improvements, Filter Robustness, and Coding Advantages for Scaled Additive Noise. - Johannes Burge; Priyank Jaini (PLoS Comput. Biol., 2017)

Deep Homogeneous Mixture Models: Representation, Separation, and Approximation. - Priyank Jaini; Pascal Poupart; Yaoliang Yu (NeurIPS, 2018)

Prometheus : Directly Learning Acyclic Directed Graph Structures for Sum-Product Networks. - Priyank Jaini; Amur Ghose; Pascal Poupart (PGM, 2018)

Sum-of-Squares Polynomial Flow. - Priyank Jaini; Kira A. Selby; Yaoliang Yu (ICML, 2019)

Tails of Triangular Flows. - Priyank Jaini; Ivan Kobyzev; Marcus A. Brubaker; Yaoliang Yu (CoRR, 2019)

Likelihood-based Density Estimation using Deep Architectures. - Priyank Jaini (2019)

Tails of Lipschitz Triangular Flows. - Priyank Jaini; Ivan Kobyzev; Yaoliang Yu; Marcus A. Brubaker (ICML, 2020)

SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows. - Didrik Nielsen; Priyank Jaini; Emiel Hoogeboom; Ole Winther; Max Welling (NeurIPS, 2020)

Complete Hierarchy of Relaxation for Constrained Signomial Positivity. - Allen Houze Wang; Priyank Jaini; Yaoliang Yu; Pascal Poupart (CoRR, 2020)

Learning directed acyclic graph SPNs in sub-quadratic time. - Amur Ghose; Priyank Jaini; Pascal Poupart (Int. J. Approx. Reason., 2020)

Sampling in Combinatorial Spaces with SurVAE Flow Augmented MCMC. - Priyank Jaini; Didrik Nielsen; Max Welling (AISTATS, 2021)

Self Normalizing Flows. - T. Anderson Keller 001; Jorn W. T. Peters; Priyank Jaini; Emiel Hoogeboom; Patrick Forré; Max Welling (ICML, 2021)

Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions. - Emiel Hoogeboom; Didrik Nielsen; Priyank Jaini; Patrick Forré; Max Welling (NeurIPS, 2021)

Learning Equivariant Energy Based Models with Equivariant Stein Variational Gradient Descent. - Priyank Jaini; Lars Holdijk; Max Welling (NeurIPS, 2021)

Argmax Flows and Multinomial Diffusion: Towards Non-Autoregressive Language Models. - Emiel Hoogeboom; Didrik Nielsen; Priyank Jaini; Patrick Forré; Max Welling (CoRR, 2021)

Particle Dynamics for Learning EBMs. - Kirill Neklyudov; Priyank Jaini; Max Welling (CoRR, 2021)

Path Integral Stochastic Optimal Control for Sampling Transition Paths. - Lars Holdijk; Yuanqi Du; Ferry Hooft; Priyank Jaini; Bernd Ensing; Max Welling (CoRR, 2022)

Recent Advancements in Tractable Probabilistic Inference (Dagstuhl Seminar 22161). - Priyank Jaini; Kristian Kersting; Antonio Vergari; Max Welling (Dagstuhl Reports, 2022)

Text-to-Image Diffusion Models are Zero Shot Classifiers. - Kevin Clark; Priyank Jaini (NeurIPS, 2023)

Stochastic Optimal Control for Collective Variable Free Sampling of Molecular Transition Paths. - Lars Holdijk; Yuanqi Du; Ferry Hooft; Priyank Jaini; Bernd Ensing; Max Welling (NeurIPS, 2023)

Intriguing Properties of Generative Classifiers. - Priyank Jaini; Kevin Clark; Robert Geirhos (ICLR, 2024)

Decoupling Semantic Similarity from Spatial Alignment for Neural Networks. - Tassilo Wald; Constantin Ulrich; Priyank Jaini; Gregor Köhler; David Zimmerer; Stefan Denner; Fabian Isensee; Michael Baumgartner 001; Klaus H. Maier-Hein (NeurIPS, 2024)

Towards flexible perception with visual memory. - Robert Geirhos; Priyank Jaini; Austin Stone; Sourabh Medapati; Xi Yi; George Toderici; Abhijit Ogale; Jonathon Shlens (CoRR, 2024)

Do generative video models learn physical principles from watching videos? - Saman Motamed; Laura Culp; Kevin Swersky; Priyank Jaini; Robert Geirhos (CoRR, 2025)

Amortized In-Context Bayesian Posterior Estimation. - Sarthak Mittal; Niels Bracher; Guillaume Lajoie; Priyank Jaini; Marcus A. Brubaker (CoRR, 2025)