Alternatives and similar repositories for jux

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

• instadeepai / flashbax
  ⚡ Flashbax: Accelerated Replay Buffers in JAX
  ☆259Updated 2 months ago
• keraJLi / rejax
  Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
  ☆244Updated 3 weeks ago
• danijar / crafter
  Benchmarking the Spectrum of Agent Capabilities
  ☆488Updated last year
• google-deepmind / dm_env
  A Python interface for reinforcement learning environments
  ☆385Updated 2 years ago
• mttga / purejaxql
  Simple single-file baselines for Q-Learning in pure-GPU setting
  ☆220Updated last week
• proroklab / popgym
  Partially Observable Process Gym
  ☆206Updated 5 months ago
• araffin / sbx
  SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms
  ☆525Updated last month
• EmptyJackson / unifloral
  Unified Implementations of Offline Reinforcement Learning Algorithms
  ☆120Updated last month
• MichalBortkiewicz / JaxGCRL
  Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.
  ☆193Updated last week
• ikostrikov / jaxrl
  JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
  ☆728Updated 3 years ago
• EdanToledo / Stoix
  🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
  ☆371Updated 3 weeks ago
• kenjyoung / MinAtar
  ☆317Updated 11 months ago
• RobertTLange / gymnax
  RL Environments in JAX 🌍
  ☆828Updated 5 months ago
• rll-research / url_benchmark
  ☆354Updated 3 years ago
• openrlbenchmark / openrlbenchmark
  ☆241Updated last year
• nico-bohlinger / RL-X
  A framework for Reinforcement Learning research.
  ☆175Updated this week
• facebookresearch / drqv2
  DrQ-v2: Improved Data-Augmented Reinforcement Learning
  ☆415Updated 3 years ago
• denisyarats / dmc2gym
  OpenAI Gym wrapper for the DeepMind Control Suite
  ☆223Updated last year
• jurgisp / pydreamer
  PyTorch implementation of DreamerV2 model-based RL algorithm
  ☆233Updated 2 years ago
• MichaelTMatthews / Craftax
  (Crafter + NetHack) in JAX. ICML 2024 Spotlight.
  ☆350Updated 4 months ago
• instadeepai / sebulba
  🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
  ☆60Updated 2 years ago
• nissymori / JAX-CORL
  Clean single-file implementation of offline RL algorithms in JAX
  ☆160Updated 10 months ago
• toshikwa / rljax
  A collection of RL algorithms written in JAX.
  ☆104Updated 3 years ago
• automl / CARL
  Benchmarking RL generalization in an interpretable way.
  ☆170Updated this week
• jurgisp / memory-maze
  Evaluating long-term memory of reinforcement learning algorithms
  ☆151Updated 2 years ago
• openai / gym3
  Vectorized interface for reinforcement learning environments
  ☆143Updated 2 years ago
• facebookresearch / dcd
  Implementations of robust Dual Curriculum Design (DCD) algorithms for unsupervised environment design.
  ☆136Updated last year
• kc-ml2 / SimpleDreamer
  A Simplified Pytorch Version of the Dreamer Algorithm
  ☆141Updated 2 years ago
• danijar / director
  Deep Hierarchical Planning from Pixels
  ☆109Updated 2 years ago
• juliusfrost / dreamer-pytorch
  Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.
  ☆317Updated last year