ExpectationMax / sklearn-jax-kernels
Composable kernels for scikit-learn implemented in JAX.
☆42Updated 4 years ago
Related projects ⓘ
Alternatives and complementary repositories for sklearn-jax-kernels
- Exponential families for JAX☆55Updated this week
- Normalizing Flows using JAX☆82Updated 11 months ago
- Jax SSM Library☆49Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆96Updated last year
- Discontinuous Hamiltonian Monte Carlo in JAX☆40Updated 4 years ago
- Graph Learning with JAX☆14Updated 2 years ago
- Inference Combinators in JAX☆44Updated 3 weeks ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆95Updated last year
- Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)☆42Updated 3 years ago
- Documentation:☆120Updated last year
- ☆78Updated 3 years ago
- ☆49Updated last year
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆21Updated 4 years ago
- Classic papers from Edwin T. Jaynes converted to latex.☆32Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆57Updated last year
- Tools for JAX☆43Updated this week
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆58Updated 4 years ago
- Tutorials and sampling algorithm comparisons☆68Updated this week
- Repository of models in Pyro☆29Updated 3 months ago
- ☆15Updated 3 years ago
- Repository for DTU Special Course, focusing on Variational Inference using Normalizing Flows (VINF). Supervised by Michael Riis Andersen☆25Updated 4 years ago
- A generic interface for linear algebra backends☆70Updated 4 months ago
- Sequential Neural Likelihood☆38Updated 5 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆41Updated last month
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆25Updated 3 years ago
- Bayesian algorithm execution (BAX)☆48Updated 3 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆64Updated last month
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 years ago
- Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend☆57Updated 5 months ago