Alternatives and similar repositories for improving-transformers-world-model-for-rl

Users that are interested in improving-transformers-world-model-for-rl are comparing it to the libraries listed below

• lucidrains / evolutionary-policy-optimization
  Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University
  ☆102Updated last month
• lucidrains / SAC-pytorch
  Implementation of Soft Actor Critic and some of its improvements in Pytorch
  ☆60Updated 9 months ago
• vmicheli / delta-iris
  Efficient World Models with Context-Aware Tokenization. ICML 2024
  ☆113Updated last year
• RyanNavillus / Syllabus
  Synchronized Curriculum Learning for RL Agents
  ☆114Updated last week
• lucidrains / x-transformers-rl
  Implementation of a transformer for reinforcement learning using `x-transformers`
  ☆69Updated last month
• chandar-lab / Recall2Imagine
  Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024
  ☆76Updated last year
• FLAIROx / jafar
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
  ☆91Updated 9 months ago
• lucidrains / scaling-vin-pytorch
  Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
  ☆37Updated last year
• lucidrains / dreamer4
  Implementation of Danijar's latest iteration for his Dreamer line of work
  ☆106Updated this week
• yilundu / ired_code_release
  ☆73Updated last year
• vladisai / PLDM
  ☆45Updated 2 weeks ago
• NanboLi / FACTS
  [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"
  ☆28Updated 5 months ago
• kvfrans / splus
  ☆119Updated 5 months ago
• facebookresearch / mtm
  MTM Masked Trajectory Models for Prediction, Representation, and Control.
  ☆159Updated 2 years ago
• facebookresearch / MRQ
  MR.Q is a general-purpose model-free reinforcement learning algorithm.
  ☆118Updated 4 months ago
• lucidrains / ppo
  An implementation of PPO in Pytorch
  ☆98Updated last week
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆110Updated last year
• lucidrains / vit-arc-slot
  Explorations into improving ViTArc with Slot Attention
  ☆43Updated last year
• radarFudan / mamba-minimal-jax
  ☆34Updated 11 months ago
• dunnolab / xland-minigrid-datasets
  XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025
  ☆81Updated 9 months ago
• rail-berkeley / SUPE
  This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."
  ☆35Updated 4 months ago
• CLAIRE-Labo / EvoTune
  Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.
  ☆118Updated 3 weeks ago
• DHDev0 / Muzero-unplugged
  Pytorch Implementation of MuZero Unplugged for gym environment. This algorithm is capable of supporting a wide range of action and observ…
  ☆34Updated 4 months ago
• mohmdelsayed / streaming-drl
  Deep reinforcement learning without experience replay, target networks, or batch updates.
  ☆271Updated 7 months ago
• DAVIAN-Robotics / SimbaV2
  Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"
  ☆76Updated last week
• weipu-zhang / STORM
  ☆114Updated last year
• seohongpark / fql
  The official implementation of flow Q-learning (FQL)
  ☆248Updated 3 months ago
• p-doom / jasmine
  A simple, performant and scalable JAX-based world modeling codebase
  ☆100Updated 2 weeks ago
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆111Updated 3 weeks ago
• luciferkonn / DT_Mem
  Thisi is the official code base for paper "Think Before You Act: Decision Transformers with Internal Working Memory"
  ☆22Updated last year