Alternatives and similar repositories for tdmpc2-jax:

Users that are interested in tdmpc2-jax are comparing it to the libraries listed below

• shacklettbp / madrona_mjx
  ☆85Updated this week
• seohongpark / ogbench
  A benchmark for offline goal-conditioned RL and offline RL
  ☆157Updated 3 weeks ago
• zhouzypaul / wsrl
  JAX implementation of WSRL and RL baselines | ICLR 2025
  ☆41Updated this week
• wertyuilife2 / bmpc
  [ICLR 2025] Bootstrapped Model Predictive Control
  ☆12Updated last week
• adityab / CrossQ
  Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"
  ☆70Updated 10 months ago
• jayeshs999 / sapg
  Code for SAPG: Split and Aggregate Policy Gradients (ICML 2024)
  ☆46Updated 7 months ago
• Improbable-AI / random-latent-exploration
  ☆23Updated 8 months ago
• MichalBortkiewicz / JaxGCRL
  Goal-Conditioned Reinforcement Learning with JAX
  ☆142Updated 3 weeks ago
• penn-pal-lab / scaffolder
  Official codebase for "Privileged Sensing Scaffolds Reinforcement Learning", contains the Scaffolder algorithm and Sensory Scaffolding Su…
  ☆27Updated last year
• Improbable-AI / pql
  Parallel Q-Learning: Scaling Off-policy Reinforcement Learning under Massively Parallel Simulation
  ☆69Updated last year
• StoneT2000 / rfcl
  (ICLR 2024) Reverse Forward Curriculum Learning
  ☆44Updated 4 months ago
• DarthUtopian / tdmpc_square_public
  Official repo for paper "TD-M(PC)^2: Improving Temporal Difference MPC Through Policy Constraint"
  ☆52Updated 2 months ago
• SonyResearch / simba
  ☆81Updated 2 months ago
• penn-pal-lab / peg
  Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.
  ☆77Updated 11 months ago
• EmptyJackson / unifloral
  Unified Implementations of Offline Reinforcement Learning Algorithms
  ☆56Updated this week
• TakuyaHiraoka / Dropout-Q-Functions-for-Doubly-Efficient-Reinforcement-Learning
  Source files to replicate experiments in my ICLR 2022 paper.
  ☆70Updated 9 months ago
• dibyaghosh / jaxrl_m
  Skeleton for scalable and flexible Jax RL implementations
  ☆80Updated last year
• aalmuzairee / dmcgb2
  Official release of the DMControl Generalization Benchmark 2 (DMC-GB2)
  ☆18Updated 10 months ago
• chauncygu / Safe-Multi-Agent-Isaac-Gym
  Safe Multi-Agent Isaac Gym benchmark for safe multi-agent reinforcement learning research.
  ☆60Updated 3 months ago
• ToruOwO / minimal-stable-PPO
  A minimal and stable PPO.
  ☆135Updated last year
• XuGW-Kevin / DrM
  DrM, a visual RL algorithm, minimizes the dormant ratio to guide exploration-exploitation trade-offs, achieving significant improvements …
  ☆73Updated 10 months ago
• imgeorgiev / DiffRL
  Learning Optimal Policies Through Contact in Differentiable Simulation
  ☆103Updated 11 months ago
• imgeorgiev / PWM
  PWM: Policy Learning with Large World Models
  ☆43Updated 2 months ago
• seohongpark / HIQL
  HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)
  ☆80Updated 4 months ago
• jrobine / twm
  Transformer-based World Models
  ☆80Updated 2 years ago
• ALRhub / d3il
  [ICLR 2024] Official implementation for "Towards Diverse Behaviors: A Benchmark for Imitation Learning with Human Demonstrations"
  ☆76Updated 2 months ago
• henrycharlesworth / dexterous-gym
  Challenging dexterous manipulation environments for RL that extend the hand manipulation environments introduced in OpenAI's Gym
  ☆44Updated 3 years ago
• mazpie / mastering-urlb
  [ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and …
  ☆37Updated last year
• ikostrikov / jaxrl2
  ☆47Updated 2 years ago
• notmahi / bet
  Code and website for Behavior Transformers: Cloning k modes with one stone.
  ☆124Updated last year