tinkoff-ai / sac-rnd
Official implementation for "Anti-Exploration by Random Network Distillation", ICML 2023
☆50Updated last year
Related projects: ⓘ
- Author's implementation of ReBRAC, a minimalist improvement upon TD3+BC☆50Updated last year
- Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)☆64Updated last year
- Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)☆38Updated last year
- Official implementation for "Q-Ensemble for Offline RL: Don't Scale the Ensemble, Scale the Batch Size", NeurIPS 2022, Offline RL Worksho…☆21Updated last year
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆46Updated last year
- ☆45Updated 5 months ago
- Official Implementation for "In-Context Reinforcement Learning from Noise Distillation"☆24Updated 7 months ago
- Code for the paper "Showing Your Offline Reinforcement Learning Work: Online Evaluation Budget Matters", ICML 2022☆28Updated 2 years ago
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning☆36Updated last month
- Official Implementation for "In-Context Reinforcement Learning for Variable Action Spaces"☆69Updated 7 months ago
- Author's implementation of ReBRAC, a minimalist improvement upon TD3+BC☆11Updated 10 months ago
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆46Updated 3 months ago
- An Open-Ended Agentic Simulator☆17Updated last month
- ☆25Updated last week
- Challenging Memory-based Deep Reinforcement Learning Agents☆76Updated 4 months ago
- Official code for ICML 2024 paper Reinformer: Max-Return Sequence Modeling for offline RL☆27Updated 3 months ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆27Updated last week
- Synthetic Experience Replay☆62Updated 3 months ago
- Accelerated replay buffers in JAX☆39Updated 2 years ago
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆19Updated 5 months ago
- Implementation (TensorFlow/keras) of the DreamerV3 model-based RL algorithm by Hafner et al. 2023☆3Updated last year
- Code for TRANSDREAMER: REINFORCEMENT LEARNING WITH TRANSFORMER WORLD MODELS☆20Updated 11 months ago
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆49Updated 8 months ago
- Fast and flexible multi-agent gridworld reinforcement learning environments.☆27Updated last month
- ☆14Updated last month
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆49Updated 11 months ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆71Updated 9 months ago
- Official implementation for "Let Offline RL Flow: Training Conservative Agents in the Latent Space of Normalizing Flows", NeurIPS 2022, O…☆12Updated last year
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- ☆11Updated 2 months ago