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
- Autoregressive Energy Machines☆78Updated 2 years ago
- Experiments for the Neural Autoregressive Flows paper☆124Updated 4 years ago
- Code for the Thermodynamic Variational Objective☆26Updated 3 years ago
- Pytorch implementation of Block Neural Autoregressive Flow☆179Updated 3 years ago
- Very deep VAEs in JAX/Flax☆46Updated 4 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- ☆80Updated 3 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆59Updated 5 years ago
- ☆180Updated 5 years ago
- Normalizing Flows using JAX☆83Updated last year
- Pytorch implementation of the Power Spherical distribution☆74Updated 11 months ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Neural likelihood-free methods in PyTorch.☆39Updated 5 years ago
- ☆42Updated 6 years ago
- BIVA: A Very Deep Hierarchy of Latent Variables forGenerative Modeling☆29Updated 5 years ago
- A Python implementation of the gradient REBAR estimator.☆46Updated 7 years ago
- PyTorch implementation of Continuously Indexed Flows paper, with many baseline normalising flows☆31Updated 3 years ago
- Train neural networks to use as SMC and importance sampling proposals☆24Updated 7 years ago
- ☆27Updated 2 years ago
- Understanding normalizing flows☆132Updated 5 years ago
- Variational Message Passing for Structured VAE (Code for ICLR 2018 paper)☆44Updated 7 years ago
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Updated 6 years ago
- ☆36Updated 2 years ago
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆26Updated 4 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
- ☆63Updated last year
- PyTorch implementation of Bidirectional Monte Carlo, Annealed Importance Sampling, and Hamiltonian Monte Carlo.☆52Updated 4 years ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆56Updated 4 years ago