Alternatives and similar repositories for jux

Users that are interested in jux are comparing it to the libraries listed below

• MichalBortkiewicz / JaxGCRL
  Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.
  ☆180Updated 3 months ago
• keraJLi / rejax
  Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
  ☆230Updated 2 months ago
• RobertTLange / gymnax
  RL Environments in JAX 🌍
  ☆792Updated 2 months ago
• EvolutionGym / evogym
  A large-scale benchmark for co-optimizing the design and control of soft robots, as seen in NeurIPS 2021.
  ☆221Updated 2 months ago
• araffin / sbx
  SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms
  ☆489Updated last week
• instadeepai / flashbax
  ⚡ Flashbax: Accelerated Replay Buffers in JAX
  ☆242Updated last week
• jurgisp / pydreamer
  PyTorch implementation of DreamerV2 model-based RL algorithm
  ☆226Updated 2 years ago
• facebookresearch / drqv2
  DrQ-v2: Improved Data-Augmented Reinforcement Learning
  ☆403Updated 3 years ago
• denisyarats / dmc2gym
  OpenAI Gym wrapper for the DeepMind Control Suite
  ☆219Updated last year
• google-deepmind / dm_env
  A Python interface for reinforcement learning environments
  ☆372Updated 2 years ago
• danijar / crafter
  Benchmarking the Spectrum of Agent Capabilities
  ☆463Updated last year
• kc-ml2 / SimpleDreamer
  A Simplified Pytorch Version of the Dreamer Algorithm
  ☆138Updated 2 years ago
• huangwl18 / modular-rl
  [ICML 2020] PyTorch Code for "One Policy to Control Them All: Shared Modular Policies for Agent-Agnostic Control"
  ☆231Updated 2 years ago
• rll-research / url_benchmark
  ☆351Updated 2 years ago
• juliusfrost / dreamer-pytorch
  Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.
  ☆305Updated last year
• openrlbenchmark / openrlbenchmark
  ☆235Updated 8 months ago
• proroklab / popgym
  Partially Observable Process Gym
  ☆196Updated 2 months ago
• openai / phasic-policy-gradient
  Code for the paper "Phasic Policy Gradient"
  ☆262Updated 2 years ago
• nico-bohlinger / RL-X
  A framework for Reinforcement Learning research.
  ☆157Updated last month
• automl / CARL
  Benchmarking RL generalization in an interpretable way.
  ☆160Updated last month
• ikostrikov / walk_in_the_park
  ☆269Updated 2 years ago
• chloechsu / revisiting-ppo
  ☆47Updated 4 years ago
• ikostrikov / jaxrl
  JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
  ☆704Updated 2 years ago
• kenjyoung / MinAtar
  ☆306Updated 7 months ago
• jurgisp / memory-maze
  Evaluating long-term memory of reinforcement learning algorithms
  ☆146Updated 2 years ago
• EdanToledo / Stoix
  🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
  ☆345Updated last week
• martius-lab / pink-noise-rl
  ☆42Updated 2 years ago
• bwfbowen / muax
  A project that provides help for using DeepMind's mctx on gym-style environments.
  ☆60Updated 9 months ago
• google-research / realworldrl_suite
  Real-World RL Benchmark Suite
  ☆356Updated 5 years ago
• openai / gym3
  Vectorized interface for reinforcement learning environments
  ☆140Updated 2 years ago