vsimkus / torch-reparametrised-mixture-distribution
PyTorch implementation of the mixture distribution family with implicit reparametrisation gradients.
☆19Updated 9 months ago
Related projects ⓘ
Alternatives and complementary repositories for torch-reparametrised-mixture-distribution
- CaDM: Context-aware Dynamics Model for Generalization in Model-based Reinforcement Learning☆63Updated 4 years ago
- Official implementation of Transformer Neural Processes☆71Updated 2 years ago
- Variational Reinforcement Learning☆16Updated 3 months ago
- This is the official source code for Sequential Neural Processes.☆40Updated last year
- Code repository complementing the ICLR 2021 paper "Unsupervised Object Keypoint Learning using Local Spatial Predictability" (https://arx…☆9Updated 3 years ago
- Generalised UDRL☆37Updated 2 years ago
- Gradient Estimation with Discrete Stein Operators (NeurIPS 2022)☆17Updated last year
- Library for Auto-Encoding Sequential Monte Carlo☆18Updated 9 months ago
- [ICLR 2022] Path integral sampler☆43Updated last year
- This repository contains implementations of the paper, Bayesian Model-Agnostic Meta-Learning.☆19Updated last year
- Clockwork Variational Autoencoder☆45Updated 3 years ago
- Codebase for "Uni[MASK]: Unified Inference in Sequential Decision Problems"☆54Updated 4 months ago
- Official data and code for our paper Systematic Evaluation of Causal Discovery in Visual Model Based Reinforcement Learning☆48Updated 3 years ago
- Reinforcement Learning with Latent Flow☆43Updated 3 years ago
- Pytorch implementation of neural processes and variants☆26Updated 3 months ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆43Updated last year
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated last year
- Trajectory-wise Multiple Choice Learning for Dynamics Generalization in Reinforcement Learning (NeurIPS 2020)☆39Updated 4 years ago
- Repository for our ICML 2019 paper: Curiosity-Bottleneck☆33Updated 2 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆20Updated 2 years ago
- ☆36Updated last year
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆31Updated last year
- Code for Goal-Aware Prediction: Learning to Model what Matters☆20Updated 4 years ago
- Structured Object-Aware Physics Prediction for Video Modeling and Planning☆32Updated 4 years ago
- Sandbox environment for generalizable agent research☆23Updated 2 years ago
- Low-variance and unbiased gradient for backpropagation through categorical random variables, with application in variational auto-encoder…☆17Updated 4 years ago
- Code for "MetaFun: Meta-Learning with Iterative Functional Updates"☆14Updated 4 years ago
- Latent World Models For Intrinsically Motivated Exploration | Official repository☆21Updated 3 years ago
- ☆41Updated 3 years ago
- Code for our NeurIPS 2020 paper Improving Generalization in Reinforcement Learning with Mixture Regularization☆32Updated 4 years ago