ericjang / nf-jax
Normalizing Flows in Jax
☆105Updated 4 years ago
Related projects ⓘ
Alternatives and complementary repositories for nf-jax
- Autoregressive Energy Machines☆77Updated last year
- Experiments for the Neural Autoregressive Flows paper☆123Updated 3 years ago
- ☆37Updated 5 years ago
- ☆178Updated 5 years ago
- Understanding normalizing flows☆131Updated 4 years ago
- Code for the Thermodynamic Variational Objective☆26Updated 2 years ago
- Pytorch implementation of Block Neural Autoregressive Flow☆176Updated 3 years ago
- ☆78Updated 3 years ago
- A Python implementation of the gradient REBAR estimator.☆45Updated 6 years ago
- BIVA: A Very Deep Hierarchy of Latent Variables forGenerative Modeling☆29Updated 5 years ago
- Very deep VAEs in JAX/Flax☆45Updated 3 years ago
- Neural likelihood-free methods in PyTorch.☆39Updated 4 years ago
- Normalizing Flows using JAX☆82Updated 11 months ago
- Code for "Differentiable Compositional Kernel Learning for Gaussian Processes" https://arxiv.org/abs/1806.04326☆71Updated 6 years ago
- simple implementation of "Improved Variational Inference with Inverse Autoregressive Flow" paper with pytorch☆53Updated 7 years ago
- PyTorch implementation of Bidirectional Monte Carlo, Annealed Importance Sampling, and Hamiltonian Monte Carlo.☆52Updated 3 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆58Updated 4 years ago
- Riemannian Convex Potential Maps☆68Updated last year
- ☆40Updated 5 years ago
- Train neural networks to use as SMC and importance sampling proposals☆24Updated 6 years ago
- Code in PyTorch for the convex combination linear IAF and the Householder Flow, J.M. Tomczak & M. Welling☆90Updated 7 years ago
- Discrete Object Generation with Reversible Inductive Construction (NeurIPS 2019)☆30Updated 3 years ago
- Implementation and tutorials of normalizing flows with the novel distributions module☆160Updated 4 years ago
- [IJCAI'19, NeurIPS'19] Anode: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEs☆104Updated 4 years ago
- Graduate topics course on learning discrete latent structure.☆66Updated 5 years ago
- PyTorch implementation of AVF☆45Updated 4 years ago
- Convex potential flows☆78Updated 3 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Updated 5 years ago
- PyTorch implementation of Continuously Indexed Flows paper, with many baseline normalising flows☆30Updated 3 years ago