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
- A small library for creating and manipulating custom JAX Pytree classes☆58Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆95Updated last year
- Jax SSM Library☆50Updated last year
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆94Updated last year
- Tools for JAX☆43Updated 2 weeks ago
- Tutorials and sampling algorithm comparisons☆68Updated this week
- Exponential families for JAX☆55Updated 2 weeks ago
- Normalizing Flows using JAX☆82Updated 11 months ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆40Updated 4 years ago
- ☆78Updated 3 years ago
- Documentation:☆120Updated last year
- Graph Learning with JAX☆14Updated 2 years ago
- Inference Combinators in JAX☆43Updated 2 weeks ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 years ago
- A generic interface for linear algebra backends☆70Updated 4 months ago
- Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)☆42Updated 3 years ago
- Classic papers from Edwin T. Jaynes converted to latex.☆32Updated 2 years ago
- Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend☆57Updated 4 months ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆41Updated 3 weeks ago
- Riemannian Convex Potential Maps☆68Updated last year
- ☆15Updated 3 years ago
- ☆29Updated 4 years ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆21Updated 4 years ago
- Differentiable and numerically stable implementation of the matrix exponential☆32Updated 4 years ago
- Pytrees + dataclasses ❤️☆61Updated 4 months ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆64Updated last month
- ☆49Updated last year
- Bayesian algorithm execution (BAX)☆46Updated 3 years ago
- This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.☆28Updated last year