probml / pgm-jaxLinks
Probabilistic Graphical Models in JAX
☆15Updated 2 years ago
Alternatives and similar repositories for pgm-jax
Users that are interested in pgm-jax are comparing it to the libraries listed below
Sorting:
- Utilities for probabilistic ML☆36Updated last year
- Bayesian Bandits☆68Updated last year
- Jax SSM Library☆49Updated 2 years ago
- ☆30Updated 3 years ago
- Library for learning and inference with Sum-product Networks utilizing TensorFlow 2.x and Keras☆48Updated 4 years ago
- Progress, Notes, Summaries and a lot of Questions on Machine Learning☆55Updated 5 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated 9 months ago
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- Pyro models and misc examples.☆19Updated 4 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- ☆29Updated last year
- ProBO: Versatile Bayesian Optimization Using Any Probabilistic Programming Language☆15Updated 6 years ago
- Python implementation of projection losses.☆27Updated 5 years ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆100Updated 2 years ago
- This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.☆32Updated last year
- Repository of models in Pyro☆29Updated 11 months ago
- ☆18Updated 3 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆60Updated 5 years ago
- Bayesian Regression Models in Pyro☆72Updated 11 months ago
- A toolbox for inference of mixture models☆16Updated 2 years ago
- Tutorials and sampling algorithm comparisons☆75Updated this week
- Reweighted Expectation Maximization☆29Updated 6 years ago
- Implicit generative models and related stuff based on the MMD, in PyTorch☆16Updated 4 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Image augmentation library for Jax☆39Updated last year
- PyTorch implementation of algorithms in https://arxiv.org/abs/2207.09238☆14Updated 2 years ago
- Loopy belief propagation for factor graphs on discrete variables in JAX☆154Updated 9 months ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆22Updated 5 years ago
- Blazingly fast implementation of the Datasaurus paper. Same Stats, Different Graphs.☆19Updated 2 years ago
- Source for experiments in the Additive Gaussian process paper, as well as extensions relating to dropout.☆22Updated 11 years ago