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 8 months ago
• DAVIAN-Robotics / SimbaV2
  Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"
  ☆85Updated 2 months ago
• EmptyJackson / unifloral
  Unified Implementations of Offline Reinforcement Learning Algorithms
  ☆195Updated last month
• facebookresearch / MRQ
  MR.Q is a general-purpose model-free reinforcement learning algorithm.
  ☆131Updated 7 months ago
• seohongpark / horizon-reduction
  The official implementation of "Horizon Reduction Makes RL Scalable"
  ☆179Updated 5 months ago
• kvfrans / rlbase_stable
  ☆44Updated last year
• MichalBortkiewicz / JaxGCRL
  Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.
  ☆232Updated 2 months ago
• mttga / purejaxql
  Simple single-file baselines for Q-Learning in pure-GPU setting
  ☆233Updated 2 months ago
• SonyResearch / simba
  ☆117Updated 11 months ago
• dibyaghosh / jaxrl_m
  Skeleton for scalable and flexible Jax RL implementations
  ☆95Updated 2 years ago
• ethanluoyc / corax
  Corax: Core RL in JAX
  ☆38Updated last year
• FLAIROx / Kinetix
  Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.
  ☆227Updated last week
• AlexGoldie / rl-learned-optimization
  Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"
  ☆30Updated last month
• UT-Austin-RPL / amago
  off-policy RL on long sequences
  ☆159Updated this week
• naumix / BiggerRegularizedOptimistic
  Official implementation of the BRO algorithm
  ☆54Updated last year
• kvfrans / fre
  Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"
  ☆57Updated last year
• RLE-Foundation / Plasticine
  Accelerating Research in Plasticity-Motivated Deep Reinforcement Learning.
  ☆35Updated last week
• ikostrikov / jaxrl2
  ☆52Updated 3 years ago
• MichaelTMatthews / Craftax_Baselines
  ☆21Updated 8 months ago
• Michael-Beukman / RobocupGym
  Reinforcement Learning inside a 3D soccer simulation
  ☆36Updated last year
• tinkoff-ai / ReBRAC
  Author's implementation of ReBRAC, a minimalist improvement upon TD3+BC
  ☆61Updated 2 years ago
• seohongpark / HILP
  Foundation Policies with Hilbert Representations (ICML 2024)
  ☆105Updated 4 months ago
• facebookresearch / minimax
  Efficient baselines for autocurricula in JAX.
  ☆206Updated last year
• InexperiencedMe / NaturalDreamer
  Simplest and Cleanest DreamerV3 implementation out there
  ☆127Updated 10 months ago
• dunnolab / xland-minigrid-datasets
  XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025
  ☆81Updated 11 months ago
• Reytuag / transformerXL_PPO_JAX
  ☆90Updated last year
• heatz123 / tldr
  Official code for TLDR: Unsupervised Goal-Conditioned RL via Temporal Distance-Aware Representations
  ☆36Updated last week
• FLAIROx / jafar
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
  ☆98Updated last year
• google-deepmind / dmc_vision_benchmark
  ☆31Updated last year
• conglu1997 / v-d4rl
  Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations
  ☆113Updated last year