ericjang / nf-jaxLinks
Normalizing Flows in Jax
☆107Updated 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:
- Autoregressive Energy Machines☆78Updated 2 years ago
- Pytorch implementation of Block Neural Autoregressive Flow☆181Updated 3 years ago
- Understanding normalizing flows☆132Updated 5 years ago
- Experiments for the Neural Autoregressive Flows paper☆125Updated 4 years ago
- Code for the Thermodynamic Variational Objective☆26Updated 3 years ago
- ☆37Updated 6 years ago
- Neural likelihood-free methods in PyTorch.☆39Updated 5 years ago
- ☆169Updated last year
- ☆80Updated 3 years ago
- Pytorch implementation of the Power Spherical distribution☆74Updated last year
- ☆180Updated 6 years ago
- Graduate topics course on learning discrete latent structure.☆67Updated 6 years ago
- [IJCAI'19, NeurIPS'19] Anode: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEs☆107Updated 4 years ago
- PyTorch implementation of Continuously Indexed Flows paper, with many baseline normalising flows☆31Updated 3 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 5 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆60Updated 5 years ago
- Code for "Differentiable Compositional Kernel Learning for Gaussian Processes" https://arxiv.org/abs/1806.04326☆71Updated 7 years ago
- Code for "Accelerating Natural Gradient with Higher-Order Invariance"☆30Updated 6 years ago
- BIVA: A Very Deep Hierarchy of Latent Variables forGenerative Modeling☆29Updated 5 years ago
- PyTorch implementation of Bidirectional Monte Carlo, Annealed Importance Sampling, and Hamiltonian Monte Carlo.☆52Updated 4 years ago
- Implementation and tutorials of normalizing flows with the novel distributions module☆168Updated 4 years ago
- Very deep VAEs in JAX/Flax☆46Updated 4 years ago
- Train neural networks to use as SMC and importance sampling proposals☆24Updated 7 years ago
- Variational Message Passing for Structured VAE (Code for ICLR 2018 paper)☆44Updated 7 years ago
- ☆59Updated 6 years ago
- ☆50Updated 4 years ago
- ☆36Updated 2 years ago
- Implementation of Spectral Inference Networks, ICLR 2019☆172Updated 6 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- a deep recurrent model for exchangeable data☆34Updated 5 years ago