Alternatives and similar repositories for jux

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

• keraJLi / rejax
  Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
  ☆258Updated 3 months ago
• instadeepai / flashbax
  ⚡ Flashbax: Accelerated Replay Buffers in JAX
  ☆269Updated 4 months ago
• automl / CARL
  Benchmarking RL generalization in an interpretable way.
  ☆173Updated 2 months ago
• mttga / purejaxql
  Simple single-file baselines for Q-Learning in pure-GPU setting
  ☆233Updated 2 months ago
• kenjyoung / MinAtar
  ☆325Updated last year
• facebookresearch / dcd
  Implementations of robust Dual Curriculum Design (DCD) algorithms for unsupervised environment design.
  ☆138Updated last year
• MichalBortkiewicz / JaxGCRL
  Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.
  ☆232Updated 2 months ago
• instadeepai / sebulba
  🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
  ☆61Updated 2 years ago
• baskuit / R-NaD
  Experimentation with Regularized Nash Dynamics on a GPU accelerated game
  ☆50Updated 2 years ago
• Farama-Foundation / AutoROM
  A tool to automate installing Atari ROMs for the Arcade Learning Environment
  ☆79Updated 5 months ago
• vwxyzjn / cleanba
  CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL
  ☆122Updated last year
• Howuhh / sac-n-jax
  Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch
  ☆56Updated 2 years ago
• martius-lab / pink-noise-rl
  ☆48Updated 2 months ago
• openrlbenchmark / openrlbenchmark
  ☆250Updated last year
• ikostrikov / jaxrl2
  ☆52Updated 3 years ago
• google-research / realworldrl_suite
  Real-World RL Benchmark Suite
  ☆364Updated 5 years ago
• Reytuag / transformerXL_PPO_JAX
  ☆90Updated last year
• facebookresearch / impact-driven-exploration
  impact-driven-exploration
  ☆133Updated 2 years ago
• jurgisp / memory-maze
  Evaluating long-term memory of reinforcement learning algorithms
  ☆163Updated 2 years ago
• proroklab / popgym
  Partially Observable Process Gym
  ☆211Updated 7 months ago
• IsaiahPressman / Kaggle_Lux_AI_2021
  ☆106Updated 4 years ago
• danijar / crafter
  Benchmarking the Spectrum of Agent Capabilities
  ☆510Updated 2 years ago
• tinker495 / jax-baseline
  Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…
  ☆62Updated last month
• EdanToledo / Stoix
  🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
  ☆386Updated 3 months ago
• bwfbowen / muax
  A project that provides help for using DeepMind's mctx on gym-style environments.
  ☆64Updated last year
• google-deepmind / dm_env
  A Python interface for reinforcement learning environments
  ☆390Updated 3 years ago
• juliusfrost / dreamer-pytorch
  Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.
  ☆321Updated 2 years ago
• openai / gym3
  Vectorized interface for reinforcement learning environments
  ☆142Updated 2 years ago
• openai / phasic-policy-gradient
  Code for the paper "Phasic Policy Gradient"
  ☆267Updated 2 years ago
• danijar / embodied
  Fast reinforcement learning research
  ☆61Updated last year