tinkoff-ai / ReBRAC
Author's implementation of ReBRAC, a minimalist improvement upon TD3+BC
☆52Updated last year
Related projects ⓘ
Alternatives and complementary repositories for ReBRAC
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning☆53Updated 3 months 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 "Q-Ensemble for Offline RL: Don't Scale the Ensemble, Scale the Batch Size", NeurIPS 2022, Offline RL Worksho…☆21Updated last year
- Official Implementation for "In-Context Reinforcement Learning from Noise Distillation"☆27Updated 2 months ago
- Code for the paper "Showing Your Offline Reinforcement Learning Work: Online Evaluation Budget Matters", ICML 2022☆28Updated 2 years ago
- Accelerated replay buffers in JAX☆40Updated 2 years ago
- ☆17Updated 5 months ago
- Open source code for paper "Optimal Goal-Reaching Reinforcement Learning via Quasimetric Learning" ICML 2023☆44Updated 6 months ago
- ☆17Updated 4 months ago
- ☆13Updated 4 months ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆38Updated 3 weeks ago
- ☆55Updated last month
- Simple JAX Graphics Library.☆23Updated 2 weeks 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
- Official code for "World Models via Policy-Guided Trajectory Diffusion", TMLR 2024☆59Updated 7 months ago
- An Open-Ended Agentic Simulator☆28Updated 3 months ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆29Updated 3 months ago
- Author's implementation of ReBRAC, a minimalist improvement upon TD3+BC☆12Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆57Updated 5 months ago
- ☆45Updated 7 months ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆52Updated 7 months ago
- Highly scalable 2D JAX physics engine.☆35Updated last week
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆40Updated this week
- ☆65Updated 2 weeks ago
- Synthetic Experience Replay☆74Updated 5 months ago
- Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"☆12Updated 3 weeks ago