coax-dev / coax

Related projects ⓘ

Alternatives and complementary repositories for coax

• henry-prior / jax-rl
  JAX implementations of core Deep RL algorithms
  ☆79Updated 2 years ago
• instadeepai / flashbax
  ⚡ Flashbax: Accelerated Replay Buffers in JAX
  ☆213Updated 3 weeks ago
• keraJLi / rejax
  ☆149Updated this week
• Farama-Foundation / Shimmy
  An API conversion tool for popular external reinforcement learning environments
  ☆139Updated last month
• araffin / sbx
  SBX: Stable Baselines Jax (SB3 + Jax)
  ☆345Updated this week
• EdanToledo / Stoix
  🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
  ☆239Updated last week
• MichaelTMatthews / Craftax
  (Crafter + NetHack) in JAX. ICML 2024 Spotlight.
  ☆208Updated last month
• RobertTLange / gymnax
  RL Environments in JAX 🌍
  ☆654Updated 4 months ago
• RobertTLange / gymnax-blines
  Baselines for gymnax 🤖
  ☆60Updated last year
• ethanluoyc / magi
  Reinforcement learning library in JAX.
  ☆102Updated last year
• chandar-lab / RLHive
  ☆100Updated 9 months ago
• automl / CARL
  Benchmarking RL generalization in an interpretable way.
  ☆133Updated 9 months ago
• kenjyoung / MinAtar
  ☆286Updated last year
• toshikwa / rljax
  A collection of RL algorithms written in JAX.
  ☆95Updated 2 years ago
• epignatelli / navix
  Accelerated minigrid environments with JAX
  ☆119Updated 3 months ago
• adaptive-intelligent-robotics / QDax
  Accelerated Quality-Diversity
  ☆268Updated last month
• juliusfrost / dreamer-pytorch
  Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.
  ☆286Updated 10 months ago
• vwxyzjn / cleanba
  CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL
  ☆105Updated 3 months ago
• ikostrikov / jaxrl
  JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
  ☆631Updated 2 years ago
• lcswillems / torch-ac
  Recurrent and multi-process PyTorch implementation of deep reinforcement Actor-Critic algorithms A2C and PPO
  ☆192Updated 2 years ago
• google-deepmind / dm_env
  A Python interface for reinforcement learning environments
  ☆346Updated last year
• rll-research / url_benchmark
  ☆332Updated 2 years ago
• google-research / realworldrl_suite
  Real-World RL Benchmark Suite
  ☆347Updated 4 years ago
• openrlbenchmark / openrlbenchmark
  ☆201Updated this week
• proroklab / popgym
  Partially Observable Process Gym
  ☆167Updated 4 months ago
• Farama-Foundation / Minari
  A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilities
  ☆297Updated this week
• google-deepmind / envlogger
  A tool for recording RL trajectories.
  ☆94Updated last week
• google-deepmind / dm_alchemy
  DeepMind Alchemy task environment: a meta-reinforcement learning benchmark
  ☆197Updated last year
• google-research / reincarnating_rl
  [NeurIPS 2022] Open source code for reusing prior computational work in RL.
  ☆91Updated last year
• dunnolab / xland-minigrid
  JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️
  ☆204Updated last week