evgenii-nikishin / omd
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
☆43Updated 3 years ago
Alternatives and similar repositories for omd:
Users that are interested in omd are comparing it to the libraries listed below
- Variational Reinforcement Learning☆16Updated 9 months ago
- Generalised UDRL☆37Updated 2 years ago
- ☆25Updated 6 years ago
- Estimating Q(s,s') with Deep Deterministic Dynamics Gradients☆32Updated 5 years ago
- A simple and easy to use implementation of the soft actor-critic algorithm.☆15Updated 2 years ago
- E2C implementation in PyTorch☆43Updated 7 years ago
- Code for reproducing experiments in Model-Based Active Exploration, ICML 2019☆78Updated 5 years ago
- Clockwork VAEs in JAX/Flax☆32Updated 3 years ago
- Implicit Normalizing Flows + Reinforcement Learning☆61Updated 5 years ago
- Sandbox environment for generalizable agent research☆25Updated 2 years ago
- Proximal Policy Optimization with Stein Control Variates:☆33Updated 7 years ago
- Public Release of Plan2vec Implementation in pyTorch☆56Updated 2 years ago
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- Solving reinforcement learning tasks which require language and vision☆32Updated 2 years ago
- krazy grid world☆25Updated 5 years ago
- Plannable Approximations to MDP Homomorphisms: Equivariance under Actions☆30Updated 4 years ago
- This repository contains code for the method and experiments of the paper "Learning with AMIGo: Adversarially Motivated Intrinsic Goals".☆61Updated last year
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)☆20Updated 3 years ago
- Invariant Causal Prediction for Block MDPs☆44Updated 4 years ago
- PyTorch Package For Quasimetric Learning☆42Updated 6 months ago
- Official pytorch implementation for our ICLR 2023 paper "Latent State Marginalization as a Low-cost Approach for Improving Exploration".☆24Updated 2 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆54Updated 2 years ago
- The implementation of "The Kanerva Machine" with Pytorch and Pyro☆12Updated 6 years ago
- ☆30Updated last year
- Revisiting Rainbow☆74Updated 3 years ago
- "CoPhy: Counterfactual Learning of Physical Dynamics", F. Baradel, N. Neverova, J. Mille, G. Mori, C. Wolf, ICLR'2020☆33Updated 5 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆21Updated 2 years ago
- Repository for the paper "Long-Horizon Visual Planning with Goal-Conditioned Hierarchical Predictors"☆44Updated 2 years ago
- Code to reproduce the results in the "Unsupervised Learning of Goal Spaces for Intrinsically Motivated Exploration"☆21Updated 7 years ago