tinkoff-ai / lb-sac
Official implementation for "Q-Ensemble for Offline RL: Don't Scale the Ensemble, Scale the Batch Size", NeurIPS 2022, Offline RL Workshop
☆21Updated last year
Related projects ⓘ
Alternatives and complementary repositories for lb-sac
- Code for the paper "Showing Your Offline Reinforcement Learning Work: Online Evaluation Budget Matters", ICML 2022☆28Updated 2 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆46Updated last year
- Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)☆37Updated last year
- Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)☆66Updated last year
- Author's implementation of ReBRAC, a minimalist improvement upon TD3+BC☆52Updated last year
- Official implementation for "Anti-Exploration by Random Network Distillation", ICML 2023☆51Updated last year
- Official implementation for "Let Offline RL Flow: Training Conservative Agents in the Latent Space of Normalizing Flows", NeurIPS 2022, O…☆12Updated last year
- Accelerated replay buffers in JAX☆39Updated 2 years ago
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning☆49Updated 3 months ago
- ☆16Updated 3 months ago
- Docker containers of baseline agents for the Crafter environment☆28Updated 2 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆14Updated last week
- Official Implementation for "In-Context Reinforcement Learning from Noise Distillation"☆27Updated last month
- Simple JAX Graphics Library.☆13Updated this week
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- on-policy optimization baselines for deep reinforcement learning☆28Updated 4 years ago
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆50Updated 5 months ago
- Code for TRANSDREAMER: REINFORCEMENT LEARNING WITH TRANSFORMER WORLD MODELS☆22Updated last year
- ☆17Updated 4 months ago
- ☆45Updated 7 months ago
- An Open-Ended Agentic Simulator☆22Updated 2 months ago
- ☆18Updated last year
- Scalable Opponent Shaping Experiments in JAX☆21Updated 6 months ago
- Implementation of Truncated Quantile Critics method for continuous reinforcement learning.☆19Updated last year
- ExORL: Exploratory Data for Offline Reinforcement Learning☆104Updated 2 years ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆40Updated last week
- Awesome In-Context RL: A curated list of In-Context Reinforcement Learning☆81Updated this week
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆52Updated 7 months ago
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆52Updated 9 months ago
- A curated list of awesome memory in reinforcement learning research materials☆18Updated 3 years ago