Alternatives and similar repositories for GradABM

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

• cambridge-mlg / sgd-gp
  Public code for running Stochastic Gradient Descent on GPs.
  ☆38Updated 3 weeks ago
• jhuggins / viabel
  Efficient, lightweight variational inference and approximation bounds
  ☆43Updated last year
• aterenin / SparseGPAX
  An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.
  ☆22Updated 4 years ago
• wesselb / lab
  A generic interface for linear algebra backends
  ☆73Updated 2 months ago
• abhiagwl / vistan
  A simple library to run variational inference on Stan models.
  ☆30Updated 2 years ago
• lockwo / distreqx
  Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
  ☆31Updated last month
• NeilGirdhar / efax
  Exponential families for JAX
  ☆69Updated 2 weeks ago
• jamesvuc / jax-bayes
  Bayesian inference with Python and Jax.
  ☆32Updated 2 years ago
• geraschenko / gmrfs
  Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)
  ☆22Updated last year
• aterenin / SparseGaussianProcesses.jl
  A Julia implementation of sparse Gaussian processes via path-wise doubly stochastic variational inference.
  ☆33Updated 4 years ago
• blackjax-devs / sampling-book
  Tutorials and sampling algorithm comparisons
  ☆75Updated this week
• google-deepmind / annealed_flow_transport
  ☆52Updated 2 years ago
• aesara-devs / aemcmc
  AeMCMC is a Python library that automates the construction of samplers for Aesara graphs representing statistical models.
  ☆39Updated last year
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• QuantEcon / lecture-jax
  Lectures on Quantitative Economics Using JAX
  ☆37Updated last week
• ae-foster / dad
  Deep Adaptive Design: Amortizing Sequential Bayesian Experimental Design
  ☆31Updated 3 years ago
• rlouf / aex
  Sampling with Blackjax on Aesara
  ☆11Updated 2 years ago
• pierrejacob / couplingsmontecarlo
  some scripts for the couplings enthusiasts!
  ☆32Updated 4 years ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆92Updated last year
• fzaiser / nonparametric-hmc
  Implementation of Nonparametric Hamiltonian Monte Carlo
  ☆12Updated 2 years ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆46Updated this week
• elizavetasemenova / prob-epi
  Course materials of "Bayesian Modelling and Probabilistic Programming with Numpyro, and Deep Generative Surrogates for Epidemiology"
  ☆64Updated 2 months ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆45Updated 4 years ago
• druidowm / OccamNet_Public
  ☆19Updated 2 years ago
• xukai92 / bsp
  ☆19Updated 3 years ago
• sash-a / CleanRL.jl
  Simple single file implementations of Reinforcement Learning algorithms in Julia
  ☆22Updated 3 months ago
• justindomke / pangolin
  probabilistic programming focused on fun
  ☆41Updated 6 months ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 4 months ago
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆49Updated last year
• YunyiShen / wild-posteriors-in-the-wild
  Posterior with interesting shapes from actually used models
  ☆13Updated 3 months ago