Fast, lightweight and parallelised simulation-based inference in JAX.
☆23Oct 28, 2025Updated 4 months ago
Alternatives and similar repositories for sbiax
Users that are interested in sbiax are comparing it to the libraries listed below
Sorting:
- Simulation-based inference in JAX☆45Nov 24, 2025Updated 3 months ago
- Implementation of Computer Vision Models in JAX (equinox)☆18Jan 15, 2026Updated last month
- madjax☆14Nov 18, 2024Updated last year
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆43Updated this week
- A JAX-based gaussian process structure learning package for time series modelling.☆37Mar 2, 2026Updated last week
- Differentiable cosmological emulators: the JAX version of CosmoPower☆44Nov 27, 2025Updated 3 months ago
- Coordinates in JAX☆29Jan 31, 2026Updated last month
- ☆24Mar 5, 2025Updated last year
- Community-sourced list of papers and resources on neural simulation-based inference.☆149Jan 28, 2026Updated last month
- Benchmarking field-level cosmological inference from galaxy surveys.☆12Jul 17, 2025Updated 7 months ago
- a differentiable lensing simulator☆10Jan 18, 2022Updated 4 years ago
- tutorials on shape measurement☆10Jan 5, 2022Updated 4 years ago
- Feynrules and UFO models for SMEFT phenomenology☆16Oct 6, 2022Updated 3 years ago
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆79Jun 20, 2025Updated 8 months ago
- Unitful Quantities in JAX☆53Updated this week
- Code for moving DM in Nbody sims and painting baryons to Nbody☆18Jan 6, 2026Updated 2 months ago
- Simulation-based (likelihood-free) inference customized for astronomical applications☆29Nov 4, 2024Updated last year
- An experiment: emcee implemented in JAX☆28Jul 1, 2022Updated 3 years ago
- Collecting research materials on neural samplers with diffusion/flow models☆59Jul 4, 2025Updated 8 months ago
- ☆36Nov 14, 2025Updated 3 months ago
- Stochastic trace estimation using JAX☆17Aug 20, 2025Updated 6 months ago
- A library for coreset algorithms, written in JAX for fast execution and GPU support.☆37Updated this week
- JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.☆33Updated this week
- Coverage tests to check the quality of your posterior estimators.☆33Aug 12, 2024Updated last year
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆32May 12, 2023Updated 2 years ago
- Probabilistic modeling of tabular data with normalizing flows.☆58Feb 11, 2026Updated 3 weeks ago
- Can we make a usable XSPEC model module in Python **experimental**☆14Oct 10, 2024Updated last year
- Pythonic class collection that helps you structure external data from LHC / HEP experiments.☆13Mar 14, 2025Updated 11 months ago
- Efficient library for spectral analysis in high-energy astrophysics.☆20Mar 2, 2026Updated last week
- Example code to speed up GP inference with gradients for high-dimensional inputs.☆14Jul 12, 2021Updated 4 years ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆256Oct 28, 2025Updated 4 months ago
- Textbook on Bayesian Data Analysis for Cognitive Science, Nicenboim, Schad, Vasishth☆22Jan 13, 2026Updated last month
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆19Jul 13, 2023Updated 2 years ago
- This is the public repository for the AbacusSummit suite, intended for specifications of the simulations and instructions for reading the…☆22Jan 29, 2026Updated last month
- ☆220Feb 21, 2026Updated 2 weeks ago
- Package for Neural Posterior Estimation and Importance Sampling focused on Astronomical Applications☆40Aug 12, 2025Updated 6 months ago
- State of the art inference for your bayesian models.☆233Jan 14, 2026Updated last month
- JAX-based linear Einstein-Boltzmann solver for cosmology☆24Nov 13, 2025Updated 3 months ago
- Python code to interface with halo catalogs and other Abacus N-body data products☆20Mar 2, 2026Updated last week