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)☆67Updated 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
- Author's implementation of ReBRAC, a minimalist improvement upon TD3+BC☆52Updated last year
- Accelerated replay buffers in JAX☆40Updated 2 years ago
- Docker containers of baseline agents for the Crafter environment☆28Updated 2 years ago
- Official Implementation for "In-Context Reinforcement Learning from Noise Distillation"☆27Updated 2 months ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- Scalable Opponent Shaping Experiments in JAX☆21Updated 7 months ago
- ☆45Updated 7 months ago
- ☆17Updated 5 months ago
- ☆18Updated last year
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆14Updated 3 weeks ago
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning☆53Updated 3 months ago
- Simple JAX Graphics Library.☆23Updated 2 weeks ago
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆55Updated 10 months ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆40Updated this week
- ☆29Updated 8 months ago
- An Open-Ended Agentic Simulator☆28Updated 3 months ago
- Code for NeurIPS 2022 paper Exploiting Reward Shifting in Value-Based Deep RL☆27Updated last year
- Vectorization techniques for fast population-based training.☆54Updated 2 years ago
- Code for TRANSDREAMER: REINFORCEMENT LEARNING WITH TRANSFORMER WORLD MODELS☆22Updated last year
- ☆13Updated 4 months ago
- A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning☆11Updated 4 years ago
- ☆28Updated last year
- ☆17Updated 4 months ago
- Official implementation of the NeurIPS 2023 paper "Discovering General Reinforcement Learning Algorithms with Adversarial Environment Des…☆22Updated 4 months ago