snu-mllab / Achievement-Distillation
Official PyTorch implementation of "Discovering Hierarchical Achievements in Reinforcement Learning via Contrastive Learning" (NeurIPS 2023)
☆27Updated last month
Related projects: ⓘ
- Guide Your Agent with Adaptive Multimodal Rewards (NeurIPS 2023 Accepted)☆32Updated 11 months ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆26Updated last month
- Official implementation of "Direct Preference-based Policy Optimization without Reward Modeling" (NeurIPS 2023)☆37Updated 2 months ago
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆49Updated 11 months ago
- ☆23Updated 10 months ago
- Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)☆49Updated 8 months ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆51Updated 5 months ago
- Codebase for "Uni[MASK]: Unified Inference in Sequential Decision Problems"☆54Updated 2 months ago
- ☆28Updated 2 years ago
- ☆14Updated 2 months ago
- Official code for "Can Wikipedia Help Offline Reinforcement Learning?" by Machel Reid, Yutaro Yamada and Shixiang Shane Gu☆100Updated 2 years ago
- ☆16Updated this week
- Implements the Messenger environment and EMMA model.☆22Updated last year
- Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)☆29Updated last year
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- Reward shaping approach for instruction following settings, leveraging language at multiple levels of abstraction.☆17Updated 3 years ago
- Official code repository for Prompt-DT.☆93Updated 2 years ago
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆21Updated last year
- CaDM: Context-aware Dynamics Model for Generalization in Model-based Reinforcement Learning☆63Updated 4 years ago
- ☆15Updated 2 years ago
- Code for the paper: Dense Reward for Free in Reinforcement Learning from Human Feedback (ICML 2024) by Alex J. Chan, Hao Sun, Samuel Holt…☆17Updated last month
- Official code for "Pretraining Representations For Data-Efficient Reinforcement Learning" (NeurIPS 2021)☆51Updated 3 years ago
- ☆19Updated 2 years ago
- ☆21Updated 8 months ago
- a simple and scalable agent for training adaptive policies with sequence-based RL☆79Updated this week
- PyTorch code accompanying the paper "Imitating Graph-Based Planning with Goal-Conditioned Policies" (ICLR 2023).☆19Updated last year
- Code for the papers Hypernetworks in Meta-Reinforcement Learning (Beck et al., 2022) and Recurrent Hypernetworks are Surprisingly Strong …☆10Updated last month
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆49Updated 8 months ago
- ☆17Updated last year
- Pytorch implementation of DreamerV2: Mastering Atari with Discrete World Models, based on the original implementation☆17Updated 2 years ago