A curated list of Decision Transformer resources (continually updated)

Code accompanying the paper Pretraining Language Models with Human Preferences

[ECCV2022] [T-PAMI] StARformer: Transformer with State-Action-Reward Representations.

The official implementation of "Transformer in Transformer as Backbone for Deep Reinforcement Learning"

Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)

Implementation of Multi-Game Decision Transformers in PyTorch

A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks

This repository is used to generate data and evaluate Decision Transformers on the CityLearn (Challenge 2022) environment for urban energy management

Retrieval-Augmented Decision Transformer: External Memory for In-context RL

stable-baselines with JAX & Haiku

Code release for the ICLR 2023 conference paper "DeFog: Decision Transformer under Random Frame Dropping"

Deep Reinforcement Learning AI to play the SuperTuxKart race game using a Decision Transformer

Code for paper Sub-goal Distillation: A Method to Improve Small Language Agents, accepted at CoLLAs 2024.

Applying regular transformer and decision transformer on solving the Rubik's cube. A paper is also written to document the results

Implementation of the decision tranformer paper in tinygrad

ATAG - Automated Test Automation Generation

some experiments with training and fine-tuning decision transformer

Pilaf is a Backgammon agent using Decision Transformers and Offline RL