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 7 months ago
• facebookresearch / MRQ
  MR.Q is a general-purpose model-free reinforcement learning algorithm.
  ☆130Updated 6 months ago
• SonyResearch / simba
  ☆116Updated 10 months ago
• seohongpark / horizon-reduction
  The official implementation of "Horizon Reduction Makes RL Scalable"
  ☆179Updated 5 months ago
• DAVIAN-Robotics / SimbaV2
  Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"
  ☆82Updated 2 months ago
• mttga / purejaxql
  Simple single-file baselines for Q-Learning in pure-GPU setting
  ☆231Updated last month
• EmptyJackson / unifloral
  Unified Implementations of Offline Reinforcement Learning Algorithms
  ☆189Updated 3 weeks ago
• dibyaghosh / jaxrl_m
  Skeleton for scalable and flexible Jax RL implementations
  ☆93Updated 2 years ago
• AlexGoldie / rl-learned-optimization
  Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"
  ☆30Updated 3 weeks ago
• heatz123 / tldr
  Official code for TLDR: Unsupervised Goal-Conditioned RL via Temporal Distance-Aware Representations
  ☆35Updated last year
• kvfrans / rlbase_stable
  ☆44Updated last year
• CognitiveAISystems / RATE
  Official implementation of Recurrent Action Transformer with Memory, an offline RL agent with memory mechanisms. https://sites.google.com…
  ☆17Updated last month
• UT-Austin-RPL / amago
  off-policy RL on long sequences
  ☆155Updated 2 weeks ago
• ethanluoyc / corax
  Corax: Core RL in JAX
  ☆38Updated last year
• marc-rigter / polygrad-world-models
  Official code for "World Models via Policy-Guided Trajectory Diffusion", TMLR 2024
  ☆72Updated last year
• regent-research / regent
  ☆28Updated 11 months ago
• enjeeneer / zero-shot-rl
  VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)
  ☆22Updated 11 months ago
• chongyi-zheng / infom
  The official implementations of Intention-conditioned Flow Occupancy Models (InFOM)
  ☆29Updated this week
• nissymori / JAX-CORL
  Clean single-file implementation of offline RL algorithms in JAX
  ☆165Updated last month
• MichaelTMatthews / Jax2D
  Highly scalable 2D JAX physics engine.
  ☆61Updated 10 months ago
• rail-berkeley / SUPE
  This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."
  ☆35Updated 6 months ago
• sukhijab / maxinforl_jax
  ☆27Updated 2 months ago
• conglu1997 / v-d4rl
  Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations
  ☆112Updated last year
• gauthamvasan / avg
  Action Value Gradient Algorithm
  ☆27Updated 7 months ago
• ikostrikov / jaxrl2
  ☆52Updated 2 years ago
• seohongpark / HILP
  Foundation Policies with Hilbert Representations (ICML 2024)
  ☆104Updated 3 months ago
• dibyaghosh / icvf_release
  Public code for "Reinforcement Learning from Passive Data via Latent Intentions"
  ☆89Updated 2 years ago
• MichaelTMatthews / Craftax_Baselines
  ☆20Updated 7 months ago
• FLAIROx / jafar
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
  ☆97Updated 11 months ago
• kvfrans / fre
  Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"
  ☆57Updated last year