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
  ☆270Updated 4 months ago
• bwfbowen / muax
  A project that provides help for using DeepMind's mctx on gym-style environments.
  ☆64Updated last year
• danielpalenicek / value_expansion
  Official code release for ICLR23 "Diminishing Return of Value Expansion Methods in Model-Based Reinforcement Learning"
  ☆16Updated 2 years ago
• kenjyoung / MinAtar
  ☆327Updated last year
• keraJLi / rejax
  Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
  ☆260Updated 3 months ago
• google-deepmind / dm_env
  A Python interface for reinforcement learning environments
  ☆391Updated 3 years ago
• kc-ml2 / SimpleDreamer
  A Simplified Pytorch Version of the Dreamer Algorithm
  ☆149Updated 2 years ago
• RobertTLange / gymnax
  RL Environments in JAX 🌍
  ☆856Updated 8 months ago
• instadeepai / sebulba
  🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
  ☆61Updated 2 years ago
• juliusfrost / dreamer-pytorch
  Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.
  ☆321Updated 2 years ago
• sotetsuk / pgx
  ♟️ Vectorized RL game environments in JAX
  ☆583Updated 11 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.
  ☆174Updated 2 months ago
• nissymori / JAX-CORL
  Clean single-file implementation of offline RL algorithms in JAX
  ☆167Updated 2 months ago
• jurgisp / memory-maze
  Evaluating long-term memory of reinforcement learning algorithms
  ☆163Updated 2 years ago
• mttga / purejaxql
  Simple single-file baselines for Q-Learning in pure-GPU setting
  ☆234Updated 2 months ago
• martius-lab / pink-noise-rl
  ☆48Updated 2 months ago
• ikostrikov / jaxrl2
  ☆52Updated 3 years ago
• ikostrikov / jaxrl
  JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
  ☆747Updated 3 years ago
• EdanToledo / Stoix
  🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
  ☆388Updated 3 months ago
• IsaiahPressman / Kaggle_Lux_AI_2021
  ☆106Updated 4 years ago
• amidos2006 / pcg_benchmark
  A benchmark to test and compare your pcg algorithm against each other
  ☆22Updated 4 months ago
• openrlbenchmark / openrlbenchmark
  ☆250Updated last year
• huangwl18 / modular-rl
  [ICML 2020] PyTorch Code for "One Policy to Control Them All: Shared Modular Policies for Agent-Agnostic Control"
  ☆232Updated 3 years ago
• EvolutionGym / evogym
  A large-scale benchmark for co-optimizing the design and control of soft robots, as seen in NeurIPS 2021.
  ☆242Updated 8 months ago
• MichaelTMatthews / Craftax
  (Crafter + NetHack) in JAX. ICML 2024 Spotlight.
  ☆371Updated 7 months ago
• rll-research / url_benchmark
  ☆359Updated 3 years ago
• danijar / director
  Deep Hierarchical Planning from Pixels
  ☆114Updated 3 years ago
• denisyarats / dmc2gym
  OpenAI Gym wrapper for the DeepMind Control Suite
  ☆227Updated last year
• andyljones / megastep
  megastep helps you build 1-million FPS reinforcement learning environments on a single GPU
  ☆146Updated 4 years ago