jax-ml / coix
Inference Combinators in JAX
☆45Updated 3 months ago
Alternatives and similar repositories for coix:
Users that are interested in coix are comparing it to the libraries listed below
- Exponential families for JAX☆60Updated last week
- ☆80Updated 3 years ago
- Classic papers from Edwin T. Jaynes converted to latex.☆32Updated 2 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆99Updated last year
- probabilistic programming focused on fun☆39Updated 3 months ago
- Tutorials and sampling algorithm comparisons☆70Updated this week
- Tools for JAX☆44Updated last week
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆64Updated 3 months ago
- A generic interface for linear algebra backends☆71Updated 7 months ago
- Composable kernels for scikit-learn implemented in JAX.☆42Updated 4 years ago
- Generic API for dispatch to Pyro backends.☆16Updated 2 years ago
- Tools for an Aesara-based PPL.☆64Updated 3 months ago
- Recognizing and exploiting conjugacy without a domain-specific language☆36Updated 5 years ago
- An introduction to functional programming for scalable statistical computing☆73Updated 6 months ago
- Prototypes of differentiable differential equation solvers in JAX.☆27Updated 4 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆38Updated 4 years ago
- A Simple Statistical Distribution Library in JAX☆16Updated 10 months ago
- Documentation:☆119Updated last year
- AeMCMC is a Python library that automates the construction of samplers for Aesara graphs representing statistical models.☆39Updated last year
- A small library for creating and manipulating custom JAX Pytree classes☆57Updated last year
- Bayesian inference for a logistic regression model in various languages☆43Updated last year
- ☆12Updated 2 years ago
- State of the art inference for your bayesian models.☆196Updated last month
- A dead simple Python package for creating custom JAX pytree objects☆62Updated 2 months ago
- ☆134Updated last month
- ☆33Updated 4 years ago
- Normalizing Flows using JAX☆82Updated last year
- All things Monte Carlo, written in JAX.☆30Updated last year
- Jax SSM Library☆49Updated 2 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 4 years ago