gensbi-examples 0.1.0

Examples for the GenSBI library

GenSBI Examples

This repository contains a collection of examples, tutorials, and recipes for GenSBI, a JAX-based library for Simulation-Based Inference using generative models.

These examples demonstrate how to use GenSBI for various tasks, including:

• Defining and running inference pipelines.
• Using different embedding networks (MLP, ResNet, etc.).
• Handling various data types (1D signals, 2D images).

Installation

Prerequisites

You need to have GenSBI and the examples package installed.

With CUDA 12 support (Recommended):

pip install gensbi[cuda12, examples]

CPU-only:

pip install gensbi

Install Examples Package

To download these examples, clone the github repository:

git clone https://github.com/aurelio-amerio/GenSBI-examples.git

Structure

• examples/: Contains standalone example scripts and notebooks.
• src/gensbi_examples: Helper utilities for the examples.

Getting Started

• My First Model:

Neural Density Estimators (NDE)

These examples demonstrate the usage of Neural Density Estimators for unconditional density estimation tasks.

• Diffusion 2D Unconditional:
• Flow Matching 2D Unconditional:
• Flow Matching 2D Unconditional (Flux1Joint):

SBI Benchmark Examples

This repository includes a comprehensive suite of Simulation-Based Inference (SBI) benchmarks. These examples cover a range of standard tasks used to evaluate SBI methods, including simple distributions, physical systems, and complex toy problems. For each task, we provide implementations using various generative models available in GenSBI, such as Flow Matching and Diffusion models with different architectures (Flux, SimFormer).

Two Moons

• Diffusion Flux:
• Diffusion Flux1Joint:
• Diffusion SimFormer:
• Flow Flux:
• Flow Flux1Joint:
• Flow SimFormer:

Bernoulli GLM

• Flow Flux:
• Flow Flux1Joint:

Gaussian Linear

• Flow Flux:
• Flow Flux1Joint:

Gaussian Mixture

• Flow Flux:
• Flow Flux1Joint:

Gravitational Waves

• GW Example:

Lensing

• Lensing Example:

SLCP

• Flow Flux:
• Flow Flux1Joint:
• Flow SimFormer: