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
  ☆20Updated 2 months ago
• EmptyJackson / unifloral
  Unified Implementations of Offline Reinforcement Learning Algorithms
  ☆88Updated 3 months ago
• facebookresearch / MRQ
  MR.Q is a general-purpose model-free reinforcement learning algorithm.
  ☆107Updated last month
• seohongpark / horizon-reduction
  The official implementation of "Horizon Reduction Makes RL Scalable"
  ☆125Updated last week
• dojeon-ai / SimbaV2
  Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"
  ☆59Updated 2 months ago
• SonyResearch / simba
  ☆103Updated 5 months ago
• naumix / BiggerRegularizedOptimistic
  Official implementation of the BRO algorithm
  ☆48Updated 6 months ago
• dibyaghosh / jaxrl_m
  Skeleton for scalable and flexible Jax RL implementations
  ☆84Updated 2 years ago
• mttga / purejaxql
  Simple single-file baselines for Q-Learning in pure-GPU setting
  ☆176Updated 4 months ago
• UT-Austin-RPL / amago
  off-policy RL on long sequences
  ☆134Updated last week
• RyanNavillus / Syllabus
  Synchronized Curriculum Learning for RL Agents
  ☆50Updated last week
• nissymori / JAX-CORL
  Clean single-file implementation of offline RL algorithms in JAX
  ☆150Updated 7 months ago
• DramaCow / jaxued
  ☆82Updated 4 months ago
• gauthamvasan / avg
  Action Value Gradient Algorithm
  ☆22Updated 2 months ago
• ethanluoyc / corax
  Corax: Core RL in JAX
  ☆38Updated last year
• MichalBortkiewicz / JaxGCRL
  Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.
  ☆179Updated 2 months ago
• FLAIROx / Kinetix
  Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.
  ☆201Updated 2 weeks ago
• MichaelTMatthews / Jax2D
  Highly scalable 2D JAX physics engine.
  ☆57Updated 5 months ago
• kvfrans / rlbase_stable
  ☆40Updated last year
• sukhijab / maxinforl_torch
  ☆44Updated 7 months ago
• jacooba / hyper
  Code for the papers Hypernetworks in Meta-Reinforcement Learning (Beck et al., 2022) and Recurrent Hypernetworks are Surprisingly Strong …
  ☆15Updated last year
• keraJLi / rejax
  Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
  ☆230Updated 2 months ago
• regent-research / regent
  ☆21Updated 6 months ago
• sukhijab / maxinforl_jax
  ☆24Updated 7 months ago
• quasimetric-learning / quasimetric-rl
  Open source code for paper "Optimal Goal-Reaching Reinforcement Learning via Quasimetric Learning" ICML 2023
  ☆46Updated 2 months ago
• AlexGoldie / rl-learned-optimization
  Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"
  ☆27Updated 3 months ago
• ShaneFlandermeyer / tdmpc2-jax
  Jax/Flax Implementation of TD-MPC2
  ☆65Updated last month
• kvfrans / fre
  Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"
  ☆57Updated last year
• marc-rigter / polygrad-world-models
  Official code for "World Models via Policy-Guided Trajectory Diffusion", TMLR 2024
  ☆66Updated last year
• keraJLi / synthetic-gymnax
  Drop-in environment replacements that make your RL algorithm train faster.
  ☆21Updated last year