amortized-inference

A Python library for simulation-based inference with deep learning

Amortized Inference for Causal Structure Learning, NeurIPS 2022

This is a community-driven collection of resources around amortized inference.

A Python library for simulation and Bayesian estimation of models with time-varying parameters.

AISTATS 2019: Reference-based Adversarial Sampling & Its applications to Soft Q-learning

ALINE: Joint Amortization for Bayesian Inference and Active Data Acquisition (Huang et al., NeurIPS 2025)

Synthetic Quantum Electrodynamics via Generative AI @ AIMS, Cape Town, May 2025

Amortized trans-dimensional inference (joint cardinality + parameters) via slot-based conditional normalizing flows — sinusoidal and Gaussian-mixture instantiations.

Lecture at ProbAI Summer School 2026

Meta-/in-context/amortized causal inference for computational identifiability

Compositional Amortized Inference for Large-Scale Hierarchical Bayesian Models

Implementing Bayesian neural networks to close the amortization gap in VAEs in PyTorch

A 2D triangle rasterizer that runs backwards, gradient-descends scene geometry, color, and opacity to match a target image.

This is the companion repository to the paper "TNFlow: Amortized Posterior Inference for Trans-Neptunian Object Surface Composition". TNFlow is a transformer + normalizing flow that inverts the Shkuratov radiative transfer model in under a second, returning full multimodal posteriors over TNO surface compositions and grain sizes.

Overcoming Selection Bias in Statistical Studies With Amortized Bayesian Inference

A tutorial on the various methods for performing approximate inference in Bayesian statistics

Official code for "Spectral Convolutional Conditional Neural Processes" (NeurIPS 2025)