Composable Effects for Flexible and Accelerated Probabilistic Programming in NumPyro

24 December 2019

Papers citing "Composable Effects for Flexible and Accelerated Probabilistic Programming in NumPyro"

50 / 143 papers shown

Title
Bayesian Hierarchical Invariant Prediction Francisco Madaleno Pernille Julie Viuff Sand Francisco C. Pereira Sergio Hernan Garrido Mejia 9 0 0 16 May 2025
JaxSGMC: Modular stochastic gradient MCMC in JAX Stephan Thaler Paul Fuchs Ana Cukarska J. Zavadlav BDL 30 0 0 16 May 2025
FlowVAT: Normalizing Flow Variational Inference with Affine-Invariant Tempering Juehang Qin Shixiao Liang C. Tunnell TPM 37 0 0 15 May 2025
Bayesian Estimation of Causal Effects Using Proxies of a Latent Interference Network Bar Weinstein Daniel Nevo CML 23 0 0 13 May 2025
HiBayES: A Hierarchical Bayesian Modeling Framework for AI Evaluation Statistics Lennart Luettgau Harry Coppock Magda Dubois Christopher Summerfield Cozmin Ududec 31 0 0 08 May 2025
Sparse mixed linear modeling with anchor-based guidance for high-entropy alloy discovery Ryo Murakami Seiji Miura Akihiro Endo Satoshi Minamoto 40 0 0 29 Apr 2025
An Image is Worth $K$ Topics: A Visual Structural Topic Model with Pretrained Image Embeddings Matías Piqueras Alexandra Segerberg Matteo Magnani Måns Magnusson Nataša Sladoje 43 0 0 14 Apr 2025
Incorporating the ChEES Criterion into Sequential Monte Carlo Samplers Andrew Millard Joshua Murphy Daniel Frisch Simon Maskell BDL 43 0 0 03 Apr 2025
DeepRV: pre-trained spatial priors for accelerated disease mapping Jhonathan Navott Daniel Jenson Seth Flaxman Elizaveta Semenova 47 0 0 27 Mar 2025
Efficiently Vectorized MCMC on Modern Accelerators Hugh Dance Pierre Glaser Peter Orbanz Ryan P. Adams 52 0 0 20 Mar 2025
Variational Autoencoded Multivariate Spatial Fay-Herriot Models Zhenhua Wang Paul A. Parker S. Holan 42 0 0 18 Mar 2025
The Architecture and Evaluation of Bayesian Neural Networks Alisa Sheinkman Sara Wade UQCV BDL 72 0 0 14 Mar 2025
Poisoning Bayesian Inference via Data Deletion and Replication Matthieu Carreau Roi Naveiro William N. Caballero AAML KELM 59 0 0 06 Mar 2025
Bayesian Optimization with Preference Exploration by Monotonic Neural Network Ensemble Hanyang Wang Juergen Branke Matthias Poloczek 99 0 0 30 Jan 2025
sbi reloaded: a toolkit for simulation-based inference workflows Jan Boelts Michael Deistler Manuel Gloeckler Álvaro Tejero-Cantero Jan-Matthis Lueckmann ... Jonas Beck J. Kapoor David S. Greenberg P. J. Gonçalves Jakob H Macke 82 4 0 26 Nov 2024
Predicting Emergent Capabilities by Finetuning Charlie Snell Eric Wallace Dan Klein Sergey Levine ELM LRM 77 5 0 25 Nov 2024
Variational Bayesian Bow tie Neural Networks with Shrinkage Alisa Sheinkman Sara Wade BDL UQCV 45 0 0 17 Nov 2024
Running Markov Chain Monte Carlo on Modern Hardware and Software Pavel Sountsov Colin Carroll Matthew D. Hoffman BDL 34 2 0 06 Nov 2024
Hamiltonian Monte Carlo Inference of Marginalized Linear Mixed-Effects Models Jinlin Lai Justin Domke Daniel Sheldon 33 0 0 31 Oct 2024
ELBOing Stein: Variational Bayes with Stein Mixture Inference Ola Rønning Eric T. Nalisnick Christophe Ley Padhraic Smyth Thomas Hamelryck BDL 52 1 0 30 Oct 2024
Flow Matching for Posterior Inference with Simulator Feedback Benjamin Holzschuh Nils Thuerey 35 0 0 29 Oct 2024
Identifying latent disease factors differently expressed in patient subgroups using group factor analysis Fabio S. Ferreira John Ashburner Arabella Bouzigues Chatrin Suksasilp Lucy L. Russell ... Fermin Moreno Barbara Borroni Samuel Kaski Jonathan D. Rohrer J. Mourão-Miranda CML 51 0 0 10 Oct 2024
Mixing of the No-U-Turn Sampler and the Geometry of Gaussian Concentration Nawaf Bou-Rabee Stefan Oberdörster 28 1 0 09 Oct 2024
Probabilistic Answer Set Programming with Discrete and Continuous Random Variables Damiano Azzolini Fabrizio Riguzzi 15 0 0 30 Sep 2024
Simulation-based inference with the Python Package sbijax Simon Dirmeier S. Ulzega Antonietta Mira Carlo Albert 38 1 0 28 Sep 2024
Automated Discovery of Pairwise Interactions from Unstructured Data Zuheng Xu Moksh Jain Ali Denton Shawn Whitfield Aniket Didolkar Berton A. Earnshaw Jason S. Hartford 29 2 0 11 Sep 2024
Active learning for regression in engineering populations: A risk-informed approach Daniel R. Clarkson L. Bull C. Wickramarachchi Elizabeth J. Cross T. Rogers Keith Worden Nikolaos Dervilis A. Hughes 18 0 0 06 Sep 2024
Inverse decision-making using neural amortized Bayesian actors Dominik Straub Tobias F. Niehues Jan Peters Constantin Rothkopf 26 0 0 04 Sep 2024
More Options for Prelabor Rupture of Membranes, A Bayesian Analysis Ashley Klein Edward Raff Elisabeth Seamon Lily Foley Timothy Bussert 19 0 0 20 Aug 2024
Flusion: Integrating multiple data sources for accurate influenza predictions E. Ray Yijin Wang Russell D. Wolfinger N. Reich 25 2 0 26 Jul 2024
SoftCVI: Contrastive variational inference with self-generated soft labels Daniel Ward Mark Beaumont Matteo Fasiolo BDL 53 0 0 22 Jul 2024
posteriordb: Testing, Benchmarking and Developing Bayesian Inference Algorithms Måns Magnusson Jakob Torgander Paul-Christian Burkner Lu Zhang Bob Carpenter Aki Vehtari 42 6 0 06 Jul 2024
Coding schemes in neural networks learning classification tasks Alexander van Meegen H. Sompolinsky 33 6 0 24 Jun 2024
Graph Structure Learning with Interpretable Bayesian Neural Networks Max Wasserman Gonzalo Mateos CML 48 6 0 20 Jun 2024
Fearless Stochasticity in Expectation Propagation Jonathan So Richard E. Turner 37 0 0 03 Jun 2024
Scalable Bayesian Learning with posteriors Samuel Duffield Kaelan Donatella Johnathan Chiu Phoebe Klett Daniel Simpson BDL UQCV 62 1 0 31 May 2024
Understanding and mitigating difficulties in posterior predictive evaluation Abhinav Agrawal Justin Domke UQCV 42 0 0 30 May 2024
Federated Learning for Non-factorizable Models using Deep Generative Prior Approximations Conor Hassan Joshua J Bon Elizaveta Semenova Antonietta Mira Kerrie Mengersen 26 0 0 25 May 2024
Dissecting the Interplay of Attention Paths in a Statistical Mechanics Theory of Transformers Lorenzo Tiberi Francesca Mignacco Kazuki Irie H. Sompolinsky 44 6 0 24 May 2024
Simulation-Based Benchmarking of Reinforcement Learning Agents for Personalized Retail Promotions Yu Xia S. Narayanamoorthy Zhengyuan Zhou Joshua Mabry OffRL 35 1 0 16 May 2024
Active Learning with Fully Bayesian Neural Networks for Discontinuous and Nonstationary Data Maxim Ziatdinov AI4CE 24 4 0 16 May 2024
Scalable Vertical Federated Learning via Data Augmentation and Amortized Inference Conor Hassan Matthew Sutton Antonietta Mira Kerrie Mengersen FedML 34 1 0 07 May 2024
GIST: Gibbs self-tuning for locally adaptive Hamiltonian Monte Carlo Nawaf Bou-Rabee Bob Carpenter Milo Marsden 43 6 0 23 Apr 2024
Diffusion posterior sampling for simulation-based inference in tall data settings J. Linhart Gabriel Victorino Cardoso Alexandre Gramfort Sylvain Le Corff Pedro L. C. Rodrigues DiffM 50 3 0 11 Apr 2024
Fusion of Gaussian Processes Predictions with Monte Carlo Sampling Marzieh Ajirak Daniel Waxman Fernando Llorente P. Djuric 19 1 0 03 Mar 2024
Demonstration of Robust and Efficient Quantum Property Learning with Shallow Shadows Hong-Ye Hu Andi Gu Swarnadeep Majumder Hang Ren Yipei Zhang Derek S. Wang Yi-Zhuang You Zlatko K. Minev S. Yelin Alireza Seif 29 24 0 27 Feb 2024
Re-Envisioning Numerical Information Field Theory (NIFTy.re): A Library for Gaussian Processes and Variational Inference G. Edenhofer Philipp Frank Jakob Roth R. Leike Massin Guerdi L. Scheel-Platz M. Guardiani Vincent Eberle M. Westerkamp T. Ensslin 28 9 0 26 Feb 2024
Stable Training of Normalizing Flows for High-dimensional Variational Inference Daniel Andrade BDL TPM 43 1 0 26 Feb 2024
Dissecting Human and LLM Preferences Junlong Li Fan Zhou Shichao Sun Yikai Zhang Hai Zhao Pengfei Liu ALM 21 5 0 17 Feb 2024
BlackJAX: Composable Bayesian inference in JAX Alberto Cabezas Adrien Corenflos Junpeng Lao Rémi Louf Antoine Carnec ... Kevin P. Murphy Juan Camilo Orduz Karm Patel Xi Wang Robert Zinkov DRL MLAU 38 18 0 16 Feb 2024