Alternatives and similar repositories for stamox

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

• JiaYaobo / fenbux
  A Simple Statistical Distribution Library in JAX
  ☆16Updated last year
• blackjax-devs / sampling-book
  Tutorials and sampling algorithm comparisons
  ☆75Updated this week
• cambridge-mlg / sgd-gp
  Public code for running Stochastic Gradient Descent on GPs.
  ☆39Updated 2 months ago
• alonfnt / bayex
  Minimal Implementation of Bayesian Optimization in JAX
  ☆95Updated 2 months ago
• QuantEcon / lecture-jax
  Lectures on Quantitative Economics Using JAX
  ☆37Updated 3 weeks ago
• patrick-kidger / wadler_lindig
  A Wadler--Lindig pretty printer for Python
  ☆39Updated 3 weeks ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆125Updated last month
• NeilGirdhar / efax
  Exponential families for JAX
  ☆72Updated 3 weeks ago
• probcomp / GenSMCP3.jl
  Automated SMC with Probabilistic Program Proposals, for the Gen PPL.
  ☆22Updated last year
• darrenjw / fp-ssc-course
  An introduction to functional programming for scalable statistical computing
  ☆75Updated last year
• jax-ml / bayeux
  State of the art inference for your bayesian models.
  ☆219Updated 2 months ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated 9 months ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆71Updated last year
• probml / JSL
  Jax SSM Library
  ☆49Updated 2 years ago
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆104Updated last year
• wesselb / lab
  A generic interface for linear algebra backends
  ☆73Updated 4 months ago
• flatironinstitute / bayes-kit
  Bayesian inference and posterior analysis for Python
  ☆44Updated last year
• jax-ml / oryx
  Oryx is a library for probabilistic programming and deep learning built on top of Jax.
  ☆269Updated this week
• acerbilab / pyvbmc
  PyVBMC: Variational Bayesian Monte Carlo algorithm for posterior and model inference in Python
  ☆120Updated 5 months ago
• darrenjw / logreg
  Bayesian inference for a logistic regression model in various languages
  ☆42Updated 2 years ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆47Updated 3 weeks ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 6 months ago
• mancusolab / traceax
  Stochastic trace estimation using JAX
  ☆15Updated 3 months ago
• ImperialCollegeLondon / csml-reading-group
  Computational statistics and machine learning reading group at Imperial College London (2019-2020)
  ☆24Updated 4 months ago
• jhuggins / viabel
  Efficient, lightweight variational inference and approximation bounds
  ☆42Updated last year
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆54Updated 2 months ago
• Joshuaalbert / jaxns
  Probabilistic Programming and Nested sampling in JAX
  ☆187Updated last week
• patrick-kidger / patdb
  A snappy + easy + pretty TUI debugger for Python.
  ☆48Updated last month
• danielward27 / flowjax
  ☆173Updated last week
• aesara-devs / aehmc
  An HMC/NUTS implementation in Aesara
  ☆31Updated 2 years ago