lucidrains / q-transformer
Implementation of Q-Transformer, Scalable Offline Reinforcement Learning via Autoregressive Q-Functions, out of Google Deepmind
☆366Updated last month
Alternatives and similar repositories for q-transformer:
Users that are interested in q-transformer are comparing it to the libraries listed below
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆520Updated 4 months ago
- ☆217Updated 3 months ago
- Implementation of Dreamer v3 in pytorch.☆508Updated 5 months ago
- Really Fast End-to-End Jax RL Implementations☆833Updated 6 months ago
- Deep reinforcement learning without experience replay, target networks, or batch updates.☆221Updated last week
- ☆267Updated 2 years ago
- Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"☆465Updated 2 weeks ago
- Code for "Learning to Model the World with Language." ICML 2024 Oral.☆379Updated last year
- General multi-task deep RL Agent☆177Updated 9 months ago
- Lamorel is a Python library designed for RL practitioners eager to use Large Language Models (LLMs).☆219Updated 4 months ago
- Code for the paper "Offline Reinforcement Learning as One Big Sequence Modeling Problem"☆484Updated 2 years ago
- ☆328Updated last year
- Online Decision Transformer☆249Updated last year
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆290Updated 2 weeks ago
- Minimal implementation of Decision Transformer: Reinforcement Learning via Sequence Modeling in PyTorch for mujoco control tasks in Open…☆262Updated 2 years ago
- Official codebase for the paper "Beyond A* Better Planning with Transformers via Search Dynamics Bootstrapping".☆362Updated 9 months ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilities☆355Updated 2 weeks ago
- ☆255Updated 2 years ago
- PyTorch implementation of DreamerV2 model-based RL algorithm☆216Updated last year
- a simple and scalable agent for training adaptive policies with sequence-based RL☆114Updated 3 weeks ago
- Multi-Agent Reinforcement Learning with JAX☆531Updated last week
- Datasets with baselines for offline multi-agent reinforcement learning.☆161Updated this week
- ☆254Updated 3 years ago
- This repository contains a collection of resources and papers on Diffusion Models for RL, accompanying the paper "Diffusion Models for Re…☆525Updated 3 months ago
- Clean baseline implementation of PPO using an episodic TransformerXL memory☆169Updated 8 months ago
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆830Updated 5 months ago
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.☆168Updated this week
- We perform functional grounding of LLMs' knowledge in BabyAI-Text☆246Updated 6 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆141Updated 3 months ago
- Mastering Diverse Domains through World Models☆1,536Updated 2 weeks ago