Alternatives and similar repositories for Kozax

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

• RobertTLange / evosax
  Evolution Strategies in JAX 🦎
  ☆613Updated last week
• adaptive-intelligent-robotics / QDax
  Accelerated Quality-Diversity
  ☆295Updated this week
• ScottAlexanderCameron / Jynx
  A neural network library written in jax
  ☆11Updated 3 months ago
• keraJLi / rejax
  Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
  ☆205Updated last month
• EMI-Group / tensorrvea
  GPU-accelerated Evolutionary Multiobjective Optimization Using Tensorized RVEA.
  ☆16Updated 2 months ago
• lasgroup / opax
  ☆18Updated 4 months ago
• resibots / pymap_elites
  map elites python reference implementation
  ☆83Updated last year
• MichalBortkiewicz / JaxGCRL
  Goal-Conditioned Reinforcement Learning with JAX
  ☆156Updated this week
• instadeepai / flashbax
  ⚡ Flashbax: Accelerated Replay Buffers in JAX
  ☆236Updated last month
• EdanToledo / Stoix
  🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
  ☆322Updated 2 weeks ago
• RobertTLange / gymnax
  RL Environments in JAX 🌍
  ☆749Updated 10 months ago
• icaros-usc / pyribs
  A bare-bones Python library for quality diversity optimization.
  ☆218Updated 2 weeks ago
• lockwo / distreqx
  Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
  ☆31Updated last month
• adaptive-intelligent-robotics / Kheperax
  High-performance JAX-powered simulator for robotic navigation in 2D mazes, optimized for Quality-Diversity algorithm research and benchma…
  ☆12Updated 7 months ago
• sses7757 / sustech-graduated-thesis
  The Typst template for SUSTech graduated student thesis
  ☆16Updated last week
• coax-dev / coax
  Modular framework for Reinforcement Learning in python
  ☆172Updated 2 years ago
• EmptyJackson / unifloral
  Unified Implementations of Offline Reinforcement Learning Algorithms
  ☆71Updated 3 weeks ago
• FLAIROx / Kinetix
  Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.
  ☆180Updated last month
• nissymori / JAX-CORL
  Clean single-file implementation of offline RL algorithms in JAX
  ☆144Updated 4 months ago
• MaximilianB2 / pc-gym
  Reinforcement learning environments for process control applications.
  ☆50Updated 2 months ago
• araffin / sbx
  SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms
  ☆443Updated this week
• instadeepai / sebulba
  🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
  ☆57Updated last year
• MichaelTMatthews / Jax2D
  Highly scalable 2D JAX physics engine.
  ☆56Updated 2 months ago
• RayZhhh / SymbolicRegressionGPU
  GP based symbolic regression framework accelerated by CUDA
  ☆9Updated 2 years ago
• nikihowe / myriad
  Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.
  ☆66Updated last year
• MushroomRL / mushroom-rl-benchmark
  Benchmarking suite for MushroomRL Deep RL algorithms
  ☆15Updated last year
• lamda-bbo / offline-moo
  Official implementation of ICML'24 paper "Offline Multi-Objective Optimization".
  ☆20Updated 7 months ago
• Probabilistic-and-Interactive-ML / awesome-plasticity-loss
  Collection of resources on plasticity loss in deep reinforcement learning
  ☆17Updated 6 months ago
• instadeepai / catx
  🐈‍⬛ Contextual bandits library for continuous action trees with smoothing in JAX
  ☆66Updated 2 years ago
• automl / CARL
  Benchmarking RL generalization in an interpretable way.
  ☆156Updated 2 months ago