scikit-learn-contrib / bdeLinks
Bayesian Deep Ensembles via MILE: easy to use, scikit-learn compatible and fast (JAX powered)
☆38Updated this week
Alternatives and similar repositories for bde
Users that are interested in bde are comparing it to the libraries listed below
Sorting:
- Gaussian processes with spherical harmonic features in JAX☆15Updated 5 months ago
- ☆16Updated 10 months ago
- Gaussian Mixture Models in Jax☆23Updated 6 months ago
- MCHMC: sampler from an arbitrary differentiable distribution☆75Updated 2 months ago
- Probabilistic Programming and Nested sampling in JAX☆219Updated last month
- State of the art inference for your bayesian models.☆231Updated 3 weeks ago
- Unleash the true power of scheduling☆33Updated 10 months ago
- The tiniest of Gaussian Process libraries☆333Updated last week
- Literature and light wrappers for gaussian process models.☆47Updated 4 years ago
- Probabilistic modeling of tabular data with normalizing flows.☆58Updated 3 weeks ago
- A lightweight and performant implementation of HMC and NUTS in Python, spun out of the PyMC project.☆59Updated last year
- Simulation-based inference in JAX☆45Updated 2 months ago
- Unitful Quantities in JAX☆52Updated last week
- A JAX-based gaussian process structure learning package for time series modelling.☆34Updated 8 months ago
- Gradient-informed particle MCMC methods☆12Updated 2 years ago
- PyMC Extras extracted from the "exoplanet" library☆19Updated 3 weeks ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆132Updated last year
- A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work☆27Updated last year
- Bayesian inference and posterior analysis for Python☆46Updated 2 years ago
- Emulate simulations easily☆109Updated this week
- Probabilistic deep learning using JAX☆15Updated last year
- Conditional density estimation with neural networks☆35Updated last year
- Finite difference tools in JAX☆23Updated 2 years ago
- Stats, linear algebra and einops for xarray☆67Updated last month
- Step size adaptation for the No-U-Turn Sampler☆26Updated 2 months ago
- A generic library for linear and non-linear Gaussian smoothing problems. The code leverages JAX and implements several linearization algo…☆12Updated last year
- Simple Xarray + JAX Integration☆22Updated 4 months ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Updated 2 years ago
- A dead simple Python package for creating custom JAX pytree objects☆66Updated last year
- Numerical quadrature with JAX☆75Updated 2 weeks ago