vsimkus / torch-reparametrised-mixture-distributionLinks
PyTorch implementation of the mixture distribution family with implicit reparametrisation gradients.
☆19Updated last year
Alternatives and similar repositories for torch-reparametrised-mixture-distribution
Users that are interested in torch-reparametrised-mixture-distribution are comparing it to the libraries listed below
Sorting:
- This is the official source code for Sequential Neural Processes.☆40Updated 2 years ago
- Generalised UDRL☆37Updated 3 years ago
- Variational Reinforcement Learning☆16Updated 10 months ago
- CaDM: Context-aware Dynamics Model for Generalization in Model-based Reinforcement Learning☆63Updated 5 years ago
- ☆51Updated 4 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆21Updated last month
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆45Updated 2 years ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated 2 years ago
- Official implementation of Transformer Neural Processes☆76Updated 2 years ago
- Official data and code for our paper Systematic Evaluation of Causal Discovery in Visual Model Based Reinforcement Learning☆48Updated 3 years ago
- [ICLR 2020, Oral] Harnessing Structures for Value-Based Planning and Reinforcement Learning☆34Updated 5 years ago
- Repository for our ICML 2019 paper: Curiosity-Bottleneck☆34Updated 2 years ago
- A PyTorch Implementation of the Importance Weighted Autoencoders☆40Updated 6 years ago
- Code for our NeurIPS 2020 paper Improving Generalization in Reinforcement Learning with Mixture Regularization☆33Updated 4 years ago
- Official Release of "Learning the Stein Discrepancy for Training and Evaluating Energy-Based Models without Sampling"☆49Updated 4 years ago
- Gradient Estimation with Discrete Stein Operators (NeurIPS 2022)☆17Updated last year
- Low-variance and unbiased gradient for backpropagation through categorical random variables, with application in variational auto-encoder…☆17Updated 4 years ago
- Code repository complementing the ICLR 2021 paper "Unsupervised Object Keypoint Learning using Local Spatial Predictability" (https://arx…☆9Updated 4 months ago
- Sandbox environment for generalizable agent research☆25Updated 2 years ago
- ☆49Updated 4 years ago
- Model-Based Visual Planning with Self-Supervised Functional Distances (ICLR 2021)☆20Updated 3 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆21Updated 2 years ago
- Library for Auto-Encoding Sequential Monte Carlo☆18Updated last year
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆35Updated 2 years ago
- Featurized Density Ratio Estimation☆20Updated 3 years ago
- [ICLR 2022] Path integral sampler☆46Updated last year
- ☆32Updated 10 months ago
- Code for "MetaFun: Meta-Learning with Iterative Functional Updates"☆14Updated 4 years ago
- Reinforcement Learning with Latent Flow☆42Updated 4 years ago
- Clockwork Variational Autoencoder☆47Updated 3 years ago