jon--lee / decision-pretrained-transformerView external linksLinks
Implemention of the Decision-Pretrained Transformer (DPT) from the paper Supervised Pretraining Can Learn In-Context Reinforcement Learning.
☆78May 28, 2024Updated last year
Alternatives and similar repositories for decision-pretrained-transformer
Users that are interested in decision-pretrained-transformer are comparing it to the libraries listed below
Sorting:
- off-policy RL on long sequences☆159Updated this week
- ☆19Jun 25, 2023Updated 2 years ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Oct 12, 2023Updated 2 years ago
- ☆20Mar 19, 2024Updated last year
- Official Implementation for "In-Context Reinforcement Learning from Noise Distillation"☆34Sep 18, 2024Updated last year
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆67Jan 18, 2024Updated 2 years ago
- iQRL: implicitly Quantized Representations for Sample-efficient Reinforcement Learning☆12Jan 8, 2025Updated last year
- Official Implementation for "In-Context Reinforcement Learning for Variable Action Spaces"☆91Feb 11, 2024Updated 2 years ago
- [ICML 2025 GenBio Workshop] Official Implementation for "Electrostatics from Laplacian Eigenbasis for Neural Network Interatomic Potentia…☆17Jun 12, 2025Updated 8 months ago
- Official code for "Can Wikipedia Help Offline Reinforcement Learning?" by Machel Reid, Yutaro Yamada and Shixiang Shane Gu☆106Jul 18, 2022Updated 3 years ago
- Measuring the Signal to Noise Ratio in Language Model Evaluation☆28Aug 19, 2025Updated 5 months ago
- Vintix: Action Model via In-Context Reinforcement Learning - - — ICML 2025☆45May 23, 2025Updated 8 months ago
- Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)☆43Aug 22, 2023Updated 2 years ago
- Official implementation of "Latent Action Learning Requires Supervision in the Presence of Distractors", ICML 2025☆33Jul 8, 2025Updated 7 months ago
- ☆24Nov 10, 2020Updated 5 years ago
- Flow RL is a high-performance RL library with flow and diffusion models.☆27Feb 6, 2026Updated last week
- Extreme Q-Learning: Max Entropy RL without Entropy☆87Feb 14, 2023Updated 3 years ago
- Foundation Policies with Hilbert Representations (ICML 2024)☆105Sep 29, 2025Updated 4 months ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 2 years ago
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025☆81Feb 13, 2025Updated last year
- Official implementation of Harnessing Mixed Offline Reinforcement Learning Datasets via Trajectory Reweighting☆17Feb 14, 2024Updated 2 years ago
- Benchmarked implementations of Offline RL Algorithms.☆76Mar 4, 2025Updated 11 months ago
- ☆31Jun 21, 2024Updated last year
- Official implementation for "Q-Ensemble for Offline RL: Don't Scale the Ensemble, Scale the Batch Size", NeurIPS 2022, Offline RL Worksho…☆21Feb 27, 2023Updated 2 years ago
- 🤖 Elegant implementations of offline safe RL algorithms in PyTorch☆231Sep 13, 2024Updated last year
- Author's PyTorch implementation of ICML'23 paper "Policy Regularization with Dataset Constraint for Offline Reinforcement Learning" for D…☆18Nov 8, 2024Updated last year
- Online Decision Transformer☆274Jan 22, 2024Updated 2 years ago
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆25May 5, 2024Updated last year
- [NeurIPS 2023] Implementation of Elastic Decision Transformer☆39Oct 12, 2023Updated 2 years ago
- A benchmark for offline goal-conditioned RL and offline RL☆331Jan 14, 2026Updated last month
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆22Dec 29, 2023Updated 2 years ago
- ☆58Feb 8, 2025Updated last year
- PyTorch implementation of Distribution Correction(DisCor) based on Soft Actor-Critic.☆38Jun 22, 2022Updated 3 years ago
- Implementation of Tactical Optimistic and Pessimistic value estimation☆25Jul 18, 2023Updated 2 years ago
- ☆10Jun 27, 2024Updated last year
- ☆22Jan 12, 2026Updated last month
- [AAAI 2023 Oral] Official code for "PiCor: Multi-Task Deep Reinforcement Learning with Policy Correction".☆20Jul 26, 2025Updated 6 months ago
- Official implementation for "Let Offline RL Flow: Training Conservative Agents in the Latent Space of Normalizing Flows", NeurIPS 2022, O…☆12Jan 31, 2023Updated 3 years ago
- ☆12May 14, 2024Updated last year