ericjang / nf-jaxLinks
Normalizing Flows in Jax
☆107Updated 5 years ago
Alternatives and similar repositories for nf-jax
Users that are interested in nf-jax are comparing it to the libraries listed below
Sorting:
- Autoregressive Energy Machines☆78Updated 3 years ago
- Code for the Thermodynamic Variational Objective☆26Updated 3 years ago
- ☆37Updated 6 years ago
- Pytorch implementation of Block Neural Autoregressive Flow☆181Updated 4 years ago
- ☆80Updated 4 years ago
- ☆172Updated last year
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 5 years ago
- Experiments for the Neural Autoregressive Flows paper☆125Updated 4 years ago
- Neural likelihood-free methods in PyTorch.☆39Updated 5 years ago
- Understanding normalizing flows☆132Updated 6 years ago
- BIVA: A Very Deep Hierarchy of Latent Variables forGenerative Modeling☆29Updated 6 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆61Updated 5 years ago
- Graduate topics course on learning discrete latent structure.☆67Updated 6 years ago
- A Python implementation of the gradient REBAR estimator.☆46Updated 7 years ago
- Code for "Accelerating Natural Gradient with Higher-Order Invariance"☆30Updated 6 years ago
- ☆181Updated 6 years ago
- Reweighted Expectation Maximization☆29Updated 6 years ago
- Code for "Differentiable Compositional Kernel Learning for Gaussian Processes" https://arxiv.org/abs/1806.04326☆71Updated 7 years ago
- ☆50Updated 5 years ago
- Train neural networks to use as SMC and importance sampling proposals☆24Updated 8 years ago
- PyTorch implementation of Continuously Indexed Flows paper, with many baseline normalising flows☆31Updated 4 years ago
- TensorFlow implementation for training MCMC samplers from the paper: Generalizing Hamiltonian Monte Carlo with Neural Network☆183Updated 7 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- PyTorch implementation of Bidirectional Monte Carlo, Annealed Importance Sampling, and Hamiltonian Monte Carlo.☆52Updated 4 years ago
- Pytorch implementation of the Power Spherical distribution☆74Updated last year
- Natural Gradient, Variational Inference☆29Updated 5 years ago
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Updated 6 years ago
- Variational Message Passing for Structured VAE (Code for ICLR 2018 paper)☆46Updated 7 years ago
- Monotone operator equilibrium networks☆54Updated 5 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆15Updated 6 years ago