Alternatives and similar repositories for sbiax

Users that are interested in sbiax are comparing it to the libraries listed below

• dirmeier / sbijax

  View on GitHub
  Simulation-based inference in JAX
  ☆45Nov 24, 2025Updated 2 months ago
• clementpoiret / Equimo

  View on GitHub
  Implementation of Computer Vision Models in JAX (equinox)
  ☆18Jan 15, 2026Updated last month
• madjax-hep / madjax

  View on GitHub
  madjax
  ☆14Nov 18, 2024Updated last year
• ChrisBoettner / gallifrey

  View on GitHub
  A JAX-based gaussian process structure learning package for time series modelling.
  ☆34May 28, 2025Updated 8 months ago
• NIFTy-PPL / NIFTy

  View on GitHub
  Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions
  ☆43Updated this week
• dpiras / cosmopower-jax

  View on GitHub
  Differentiable cosmological emulators: the JAX version of CosmoPower
  ☆43Nov 27, 2025Updated 2 months ago
• GalacticDynamics / coordinax

  View on GitHub
  Coordinates in JAX
  ☆29Jan 31, 2026Updated 2 weeks ago
• ViniciusMikuni / OmniLearn

  View on GitHub
  ☆23Mar 5, 2025Updated 11 months ago
• smsharma / awesome-neural-sbi

  View on GitHub
  Community-sourced list of papers and resources on neural simulation-based inference.
  ☆147Jan 28, 2026Updated 2 weeks ago
• VMBoehm / MADLens

  View on GitHub
  a differentiable lensing simulator
  ☆10Jan 18, 2022Updated 4 years ago
• beckermr / shape-measurement-tutorials

  View on GitHub
  tutorials on shape measurement
  ☆10Jan 5, 2022Updated 4 years ago
• hsimonfroy / benchmark-field-level

  View on GitHub
  Benchmarking field-level cosmological inference from galaxy surveys.
  ☆12Jul 17, 2025Updated 7 months ago
• SMEFTsim / SMEFTsim

  View on GitHub
  Feynrules and UFO models for SMEFT phenomenology
  ☆16Oct 6, 2022Updated 3 years ago
• mackelab / simformer

  View on GitHub
  Combination of transformers and diffusion models for flexible all-in-one simulation-based inference
  ☆79Jun 20, 2025Updated 7 months ago
• DhayaaAnbajagane / BaryonForge

  View on GitHub
  Code for moving DM in Nbody sims and painting baryons to Nbody
  ☆18Jan 6, 2026Updated last month
• google-deepmind / xarray_jax

  View on GitHub
  ☆32Nov 14, 2025Updated 3 months ago
• dfm / emcee-jax

  View on GitHub
  An experiment: emcee implemented in JAX
  ☆28Jul 1, 2022Updated 3 years ago
• wangbingjie / sbi_pp

  View on GitHub
  Simulation-based (likelihood-free) inference customized for astronomical applications
  ☆29Nov 4, 2024Updated last year
• J-zin / Awesome-Diffusion-Flow-Samplers

  View on GitHub
  Collecting research materials on neural samplers with diffusion/flow models
  ☆59Jul 4, 2025Updated 7 months ago
• gchq / coreax

  View on GitHub
  A library for coreset algorithms, written in JAX for fast execution and GPU support.
  ☆37Feb 10, 2026Updated last week
• Ciela-Institute / tarp

  View on GitHub
  Coverage tests to check the quality of your posterior estimators.
  ☆33Aug 12, 2024Updated last year
• mancusolab / traceax

  View on GitHub
  Stochastic trace estimation using JAX
  ☆17Aug 20, 2025Updated 5 months ago
• GalSim-developers / JAX-GalSim

  View on GitHub
  JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.
  ☆33Updated this week
• tomcharnock / IMNN

  View on GitHub
  Using neural networks to extract sufficient statistics from data by maximising the Fisher information
  ☆32May 12, 2023Updated 2 years ago
• wcxve / elisa

  View on GitHub
  Efficient library for spectral analysis in high-energy astrophysics.
  ☆20Updated this week
• jfcrenshaw / pzflow

  View on GitHub
  Probabilistic modeling of tabular data with normalizing flows.
  ☆58Updated this week
• cxcsds / xspec-models-cxc

  View on GitHub
  Can we make a usable XSPEC model module in Python **experimental**
  ☆14Oct 10, 2024Updated last year
• riga / order

  View on GitHub
  Pythonic class collection that helps you structure external data from LHC / HEP experiments.
  ☆13Mar 14, 2025Updated 11 months ago
• fidero / gp-derivative

  View on GitHub
  Example code to speed up GP inference with gradients for high-dimensional inputs.
  ☆14Jul 12, 2021Updated 4 years ago
• kazewong / flowMC

  View on GitHub
  Normalizing-flow enhanced sampling package for probabilistic inference in Jax
  ☆255Oct 28, 2025Updated 3 months ago
• bnicenboim / bayescogsci

  View on GitHub
  Textbook on Bayesian Data Analysis for Cognitive Science, Nicenboim, Schad, Vasishth
  ☆21Jan 13, 2026Updated last month
• mackelab / gatsbi

  View on GitHub
  GATSBI: Generative Adversarial Training for Simulation-Based Inference
  ☆19Jul 13, 2023Updated 2 years ago
• abacusorg / AbacusSummit

  View on GitHub
  This is the public repository for the AbacusSummit suite, intended for specifications of the simulations and instructions for reading the…
  ☆22Jan 29, 2026Updated 2 weeks ago
• abacusorg / abacusutils

  View on GitHub
  Python code to interface with halo catalogs and other Abacus N-body data products
  ☆20Feb 2, 2026Updated 2 weeks ago
• danielward27 / flowjax

  View on GitHub
  ☆218Feb 1, 2026Updated 2 weeks ago
• kmzzhang / nbi

  View on GitHub
  Package for Neural Posterior Estimation and Importance Sampling focused on Astronomical Applications
  ☆39Aug 12, 2025Updated 6 months ago
• jax-ml / bayeux

  View on GitHub
  State of the art inference for your bayesian models.
  ☆232Jan 14, 2026Updated last month
• ohahn / DISCO-EB

  View on GitHub
  JAX-based linear Einstein-Boltzmann solver for cosmology
  ☆24Nov 13, 2025Updated 3 months ago
• DifferentiableUniverseInitiative / jax_cosmo

  View on GitHub
  A differentiable cosmology library in JAX
  ☆223Jun 27, 2025Updated 7 months ago