ericjang / nf-jaxLinks
Normalizing Flows in Jax
☆108Updated 4 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:
- ☆37Updated 5 years ago
- Experiments for the Neural Autoregressive Flows paper☆124Updated 3 years ago
- Autoregressive Energy Machines☆78Updated 2 years ago
- Code for the Thermodynamic Variational Objective☆26Updated 2 years ago
- Neural likelihood-free methods in PyTorch.☆39Updated 5 years ago
- ☆180Updated 5 years ago
- Understanding normalizing flows☆132Updated 5 years ago
- Pytorch implementation of Block Neural Autoregressive Flow☆178Updated 3 years ago
- BIVA: A Very Deep Hierarchy of Latent Variables forGenerative Modeling☆29Updated 5 years ago
- ☆80Updated 3 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆59Updated 5 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Normalizing Flows using JAX☆83Updated last year
- Pytorch implementation of the Power Spherical distribution☆74Updated 10 months ago
- Very deep VAEs in JAX/Flax☆46Updated 3 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- Code for "A Spectral Approach to Gradient Estimation for Implicit Distributions" (ICML'18)☆33Updated 2 years ago
- PyTorch implementation of Bidirectional Monte Carlo, Annealed Importance Sampling, and Hamiltonian Monte Carlo.☆52Updated 4 years ago
- ☆42Updated 6 years ago
- PyTorch implementation of Continuously Indexed Flows paper, with many baseline normalising flows☆31Updated 3 years ago
- Code for "Functional variational Bayesian neural networks" (https://arxiv.org/abs/1903.05779)☆84Updated 4 years ago
- A Python implementation of the gradient REBAR estimator.☆46Updated 6 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆87Updated 2 years ago
- Reweighted Expectation Maximization☆29Updated 5 years ago
- Code for "Differentiable Compositional Kernel Learning for Gaussian Processes" https://arxiv.org/abs/1806.04326☆71Updated 6 years ago
- Graduate topics course on learning discrete latent structure.☆67Updated 6 years ago
- simple implementation of "Improved Variational Inference with Inverse Autoregressive Flow" paper with pytorch☆52Updated 7 years ago
- ☆168Updated 10 months ago
- A differentiation API for PyTorch☆30Updated 4 years ago
- ☆22Updated 4 years ago