Alternatives and similar repositories for tdmpc2-jax

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

• wertyuilife2 / bmpc
  [ICLR 2025] Bootstrapped Model Predictive Control
  ☆21Updated 2 months ago
• Improbable-AI / pql
  Parallel Q-Learning: Scaling Off-policy Reinforcement Learning under Massively Parallel Simulation
  ☆75Updated 2 years ago
• StoneT2000 / rfcl
  (ICLR 2024) Reverse Forward Curriculum Learning
  ☆49Updated 10 months ago
• adityab / CrossQ
  Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"
  ☆79Updated last year
• MichalBortkiewicz / JaxGCRL
  Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.
  ☆188Updated 4 months ago
• Viraj-Joshi / MTBench
  ☆32Updated last month
• seohongpark / ogbench
  A benchmark for offline goal-conditioned RL and offline RL
  ☆248Updated 3 months ago
• penn-pal-lab / scaffolder
  Official codebase for "Privileged Sensing Scaffolds Reinforcement Learning", contains the Scaffolder algorithm and Sensory Scaffolding Su…
  ☆30Updated last year
• dibyaghosh / jaxrl_m
  Skeleton for scalable and flexible Jax RL implementations
  ☆87Updated 2 years ago
• SonyResearch / simba
  ☆108Updated 7 months ago
• shacklettbp / madrona_mjx
  ☆136Updated 2 months ago
• DarthUtopian / tdmpc_square_public
  Official repo for paper "TD-M(PC)^2: Improving Temporal Difference MPC Through Policy Constraint"
  ☆67Updated 7 months ago
• imgeorgiev / PWM
  PWM: Policy Learning with Large World Models
  ☆57Updated 2 months ago
• zhouzypaul / wsrl
  JAX implementation of WSRL and RL baselines | ICLR 2025
  ☆112Updated 2 months ago
• aalmuzairee / dmcgb2
  Official release of the DMControl Generalization Benchmark 2 (DMC-GB2)
  ☆22Updated 2 months ago
• mahaitongdae / diffusion_policy_online_rl
  ☆28Updated 2 months ago
• TakuyaHiraoka / Dropout-Q-Functions-for-Doubly-Efficient-Reinforcement-Learning
  Source files to replicate experiments in my ICLR 2022 paper.
  ☆70Updated 2 months ago
• ikostrikov / rlpd
  ☆341Updated 2 years ago
• penn-pal-lab / peg
  Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.
  ☆78Updated last year
• ToruOwO / minimal-stable-PPO
  A minimal and stable PPO.
  ☆143Updated last year
• seohongpark / horizon-reduction
  The official implementation of "Horizon Reduction Makes RL Scalable"
  ☆139Updated 2 months ago
• seohongpark / HILP
  Foundation Policies with Hilbert Representations (ICML 2024)
  ☆97Updated last week
• Improbable-AI / random-latent-exploration
  ☆25Updated last year
• nico-bohlinger / RL-X
  A framework for Reinforcement Learning research.
  ☆161Updated last month
• nissymori / JAX-CORL
  Clean single-file implementation of offline RL algorithms in JAX
  ☆156Updated 9 months ago
• ALRhub / d3il
  [ICLR 2024] Official implementation for "Towards Diverse Behaviors: A Benchmark for Imitation Learning with Human Demonstrations"
  ☆85Updated 7 months ago
• dojeon-ai / SimbaV2
  Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"
  ☆66Updated 4 months ago
• imgeorgiev / DiffRL
  Learning Optimal Policies Through Contact in Differentiable Simulation
  ☆105Updated last year
• youngwoon / robot-learning
  ☆53Updated 2 years ago
• jayeshs999 / sapg
  Code for SAPG: Split and Aggregate Policy Gradients (ICML 2024)
  ☆55Updated last year