Alternatives and similar repositories for brax-trainer

Users that are interested in brax-trainer are comparing it to the libraries listed below

• google-deepmind / lm_act
  LMAct: A Benchmark for In-Context Imitation Learning with Long Multimodal Demonstrations
  ☆25Updated 6 months ago
• EmptyJackson / unifloral
  Unified Implementations of Offline Reinforcement Learning Algorithms
  ☆125Updated last month
• seohongpark / horizon-reduction
  The official implementation of "Horizon Reduction Makes RL Scalable"
  ☆161Updated 3 months ago
• DAVIAN-Robotics / SimbaV2
  Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"
  ☆77Updated 3 weeks ago
• SonyResearch / simba
  ☆112Updated 9 months ago
• quasimetric-learning / quasimetric-rl
  Open source code for paper "Optimal Goal-Reaching Reinforcement Learning via Quasimetric Learning" ICML 2023
  ☆50Updated 6 months ago
• naumix / BiggerRegularizedOptimistic
  Official implementation of the BRO algorithm
  ☆51Updated 10 months ago
• kvfrans / rlbase_stable
  ☆43Updated last year
• heatz123 / tldr
  Official code for TLDR: Unsupervised Goal-Conditioned RL via Temporal Distance-Aware Representations
  ☆34Updated last year
• mttga / purejaxql
  Simple single-file baselines for Q-Learning in pure-GPU setting
  ☆220Updated 2 weeks ago
• RLE-Foundation / Plasticine
  Accelerating Research in Plasticity-Motivated Deep Reinforcement Learning.
  ☆30Updated 5 months ago
• ShaneFlandermeyer / tdmpc2-jax
  Jax/Flax Implementation of TD-MPC2
  ☆66Updated this week
• rail-berkeley / SUPE
  This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."
  ☆35Updated 4 months ago
• ethanluoyc / corax
  Corax: Core RL in JAX
  ☆38Updated last year
• dibyaghosh / jaxrl_m
  Skeleton for scalable and flexible Jax RL implementations
  ☆92Updated 2 years ago
• gauthamvasan / avg
  Action Value Gradient Algorithm
  ☆25Updated 6 months ago
• seohongpark / HILP
  Foundation Policies with Hilbert Representations (ICML 2024)
  ☆102Updated 2 months ago
• MichaelTMatthews / Craftax_Baselines
  ☆19Updated 6 months ago
• DramaCow / jaxued
  ☆87Updated 2 months ago
• nissymori / JAX-CORL
  Clean single-file implementation of offline RL algorithms in JAX
  ☆160Updated 11 months ago
• UT-Austin-RPL / amago
  off-policy RL on long sequences
  ☆149Updated 3 months ago
• seohongpark / METRA
  METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)
  ☆79Updated 2 years ago
• AlexGoldie / rl-learned-optimization
  Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"
  ☆29Updated 7 months ago
• MichalBortkiewicz / JaxGCRL
  Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.
  ☆193Updated this week
• enjeeneer / zero-shot-rl
  VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)
  ☆21Updated 10 months ago
• Reytuag / transformerXL_PPO_JAX
  ☆87Updated last year
• FLAIROx / Kinetix
  Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.
  ☆219Updated 3 weeks ago
• tinkoff-ai / ReBRAC
  Author's implementation of ReBRAC, a minimalist improvement upon TD3+BC
  ☆60Updated 2 years ago
• sukhijab / maxinforl_jax
  ☆27Updated last month
• facebookresearch / MRQ
  MR.Q is a general-purpose model-free reinforcement learning algorithm.
  ☆123Updated 5 months ago