evgenii-nikishin / omd
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
☆43Updated 3 years ago
Related projects: ⓘ
- Estimating Q(s,s') with Deep Deterministic Dynamics Gradients☆30Updated 4 years ago
- Code for the paper "Gamma-Models: Generative Temporal Difference Learning for Infinite-Horizon Prediction"☆39Updated last year
- Code for reproducing experiments in Model-Based Active Exploration, ICML 2019☆77Updated 5 years ago
- Variational Reinforcement Learning☆16Updated last month
- Simplistic Pytorch Implementation of the Dreamer-RL☆20Updated 2 years ago
- ☆26Updated 5 years ago
- Generalised UDRL☆37Updated 2 years ago
- Official pytorch implementation for our ICLR 2023 paper "Latent State Marginalization as a Low-cost Approach for Improving Exploration".☆24Updated last year
- Implicit Normalizing Flows + Reinforcement Learning☆60Updated 5 years ago
- This repository contains code for the method and experiments of the paper "Learning with AMIGo: Adversarially Motivated Intrinsic Goals".☆61Updated last year
- Proximal Policy Optimization with Stein Control Variates:☆33Updated 6 years ago
- Docker containers of baseline agents for the Crafter environment☆27Updated 2 years ago
- A simple and easy to use implementation of the soft actor-critic algorithm.☆15Updated 2 years ago
- Repository for the paper "Long-Horizon Visual Planning with Goal-Conditioned Hierarchical Predictors"☆44Updated last year
- Clockwork VAEs in JAX/Flax☆31Updated 3 years ago
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the …☆83Updated 3 years ago
- Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)☆20Updated 2 years ago
- Sandbox environment for generalizable agent research☆22Updated 2 years ago
- krazy grid world☆25Updated 4 years ago
- Implementation for ICML 2019 paper, EMI: Exploration with Mutual Information.☆37Updated 3 years ago
- 📴 OffCon^3: SOTA PyTorch SAC and TD3 Implementations (arxiv: 2101.11331)☆24Updated 3 years ago
- Code to reproduce the experiments in The Mirage of Action-Dependent Baselines in Reinforcement Learning.☆17Updated 6 years ago
- Revisiting Rainbow☆73Updated 3 years ago
- P3O paper code☆26Updated 5 years ago
- Code Release for Task Agnostic Dynamics Priors for Deep Reinforcement Learning☆12Updated 5 years ago
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- Continual Reinforcement Learning in 3D Non-stationary Environments☆35Updated 5 years ago
- ☆34Updated last year
- Autoregressive policies for continuous control reinforcement learning☆28Updated 5 years ago
- Vectorization techniques for fast population-based training.☆52Updated 2 years ago