jamesvuc / jax-bayes
Bayesian inference with Python and Jax.
☆32Updated 2 years ago
Alternatives and similar repositories for jax-bayes:
Users that are interested in jax-bayes are comparing it to the libraries listed below
- ☆49Updated 2 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆102Updated last year
- ☆80Updated 3 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆89Updated 2 months ago
- Riemannian Optimization Using JAX☆48Updated last year
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆26Updated last month
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆41Updated 2 weeks ago
- Normalizing Flows using JAX☆83Updated last year
- A generic interface for linear algebra backends☆71Updated 3 weeks ago
- Turn jitted jax functions back into python source code☆22Updated 3 months ago
- ☆151Updated 2 weeks ago
- Exponential families for JAX☆63Updated this week
- Turning SymPy expressions into JAX functions☆44Updated 4 years ago
- ☆19Updated 2 years ago
- Tutorial materials of the Probabilistic Numerics Spring School.☆34Updated last year
- JAX Arrays for human consumption☆90Updated last year
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- Website for the book "The Elements of Differentiable Programming".☆13Updated 6 months ago
- Public code for running Stochastic Gradient Descent on GPs.☆36Updated 4 months ago
- Multiple dispatch over abstract array types in JAX.☆115Updated last month
- Nonparametric Differential Equation Modeling☆53Updated last year
- Code for efficiently sampling functions from GP(flow) posteriors☆70Updated 4 years ago
- Tools for JAX☆45Updated this week
- ☆15Updated 2 years ago
- ☆30Updated 6 months ago
- Matrix-free linear algebra in JAX.☆116Updated 2 months ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆24Updated 2 years ago
- Extensible Tensorflow library for differentiable particle filtering. ICML 2021.☆41Updated 2 years ago
- Pytorch-like dataloaders for JAX.☆76Updated 5 months ago