bramgrooten / mask-distractions
[AAMAS 2024] Code for the paper "MaDi: Learning to Mask Distractions for Generalization in Visual Deep RL"
☆25Updated last month
Alternatives and similar repositories for mask-distractions:
Users that are interested in mask-distractions are comparing it to the libraries listed below
- A framework for evaluating LLMs in Atari games☆15Updated 2 weeks ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆70Updated 11 months ago
- ☆47Updated 2 years ago
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆19Updated last year
- ☆42Updated 2 years ago
- Gridworld domains in the gym interface☆27Updated 7 months ago
- Goal-Conditioned Reinforcement Learning with JAX☆153Updated last week
- LAMBDA is a model-based reinforcement learning agent that uses Bayesian world models for safe policy optimization☆33Updated 2 years ago
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Updated last year
- ☆23Updated 2 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆52Updated last year
- Implementation of Tactical Optimistic and Pessimistic value estimation☆25Updated last year
- Conservative Q learning in Jax☆53Updated 2 years ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆82Updated 5 months ago
- ☆44Updated 5 months ago
- Official release of the DMControl Generalization Benchmark 2 (DMC-GB2)☆19Updated 11 months ago
- ☆24Updated last year
- ReDo: The Dormant Neuron Phenomenon in Deep Reinforcement Learning (pytorch)☆26Updated 6 months ago
- Unified Implementations of Offline Reinforcement Learning Algorithms☆67Updated 2 weeks ago
- Benchmarking RL generalization in an interpretable way.☆155Updated 2 months ago
- A version of the DeepMind Control Suite with randomly generated graphics, for measuring visual generalization in continuous control.☆18Updated 4 years ago
- [ICLR 2023 Oral] The official implementation of SQL and EQL in "Offline RL with No OOD Actions: In-Sample Learning via Implicit Value Reg…☆43Updated last year
- Code for the NeurIPS 2021 paper "Safe Reinforcement Learning by Imagining the Near Future"☆44Updated 3 years ago
- EARL: Environment for Autonomous Reinforcement Learning☆37Updated 2 years ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆80Updated 2 years ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆113Updated 3 years ago
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆100Updated 11 months ago
- [NeurIPS 2023] The official implementation of "Offline Multi-Agent Reinforcement Learning with Implicit Global-to-Local Value Regularizat…☆34Updated last year
- Evaluating long-term memory of reinforcement learning algorithms☆142Updated last year
- [ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and …☆38Updated last year