InexperiencedMe / NaturalDreamer
Simplest and Cleanest DreamerV3 implementation out there
☆57Updated 3 weeks ago
Alternatives and similar repositories for NaturalDreamer:
Users that are interested in NaturalDreamer are comparing it to the libraries listed below
- MR.Q is a general-purpose model-free reinforcement learning algorithm.☆85Updated last week
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆42Updated this week
- Simple single-file baselines for Q-Learning in pure-GPU setting☆151Updated 3 weeks ago
- Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"☆22Updated 4 months ago
- ☆76Updated 3 weeks ago
- Highly scalable 2D JAX physics engine.☆53Updated last month
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - —☆66Updated 2 months ago
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.☆172Updated 3 weeks ago
- ☆80Updated last month
- ☆74Updated 5 months ago
- ☆20Updated 9 months ago
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆64Updated 10 months ago
- Goal-Conditioned Reinforcement Learning with JAX☆142Updated 2 weeks ago
- Challenging Memory-based Deep Reinforcement Learning Agents☆97Updated 5 months ago
- ☆197Updated 2 weeks ago
- ☆41Updated 9 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆72Updated 7 months ago
- Code for the paper "Harnessing Discrete Representations for Continual Reinforcement Learning"☆14Updated 10 months ago
- ☆18Updated 2 months ago
- JAX implementation of RL algorithms and vectorized environments☆43Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆99Updated last year
- Accelerated minigrid environments with JAX☆133Updated 8 months ago
- Foundation Policies with Hilbert Representations (ICML 2024)☆82Updated last year
- Simple JAX Graphics Library.☆36Updated 5 months ago
- a simple and scalable agent for training adaptive policies with sequence-based RL☆118Updated last week
- ☆72Updated last year
- Efficient baselines for autocurricula in JAX.☆186Updated 7 months ago
- ☆29Updated last month
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆52Updated last year
- Code for TRANSDREAMER: REINFORCEMENT LEARNING WITH TRANSFORMER WORLD MODELS☆25Updated last year