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 🦎
  ☆718Updated 4 months ago
• EdanToledo / Stoix
  🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
  ☆384Updated 2 months ago
• keraJLi / rejax
  Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
  ☆257Updated 2 months ago
• RobertTLange / gymnax
  RL Environments in JAX 🌍
  ☆848Updated 7 months ago
• adaptive-intelligent-robotics / QDax
  Accelerated Quality-Diversity
  ☆335Updated 2 months ago
• instadeepai / flashbax
  ⚡ Flashbax: Accelerated Replay Buffers in JAX
  ☆269Updated 4 months ago
• hvarfner / vanilla_bo_in_highdim
  ☆35Updated 11 months ago
• icaros-usc / pyribs
  A bare-bones Python library for quality diversity optimization.
  ☆251Updated last month
• thomaspinder / GPJax
  Gaussian processes in JAX and Flax.
  ☆576Updated last week
• dso-org / deep-symbolic-optimization
  A deep learning framework for symbolic optimization.
  ☆704Updated this week
• araffin / sbx
  SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms
  ☆549Updated this week
• bwfbowen / muax
  A project that provides help for using DeepMind's mctx on gym-style environments.
  ☆64Updated last year
• google-deepmind / kfac-jax
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.
  ☆309Updated last week
• lasgroup / opax
  ☆19Updated last year
• lamda-bbo / offline-moo
  Official implementation of ICML'24 paper "Offline Multi-Objective Optimization".
  ☆24Updated 7 months ago
• danielward27 / flowjax
  ☆211Updated last week
• luchris429 / purejaxrl
  Really Fast End-to-End Jax RL Implementations
  ☆1,013Updated last year
• epignatelli / navix
  Accelerated minigrid environments with JAX
  ☆156Updated 3 months ago
• CSML-IIT-UCL / linear_operator_learning
  A package to Learn Linear Operators
  ☆14Updated last month
• cavalab / srbench
  A living benchmark framework for symbolic regression
  ☆292Updated 3 months ago
• instadeepai / jumanji
  🕹️ A diverse suite of scalable reinforcement learning environments in JAX
  ☆798Updated last month
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆531Updated last month
• smorad / memax
  Deep memory and sequence models in JAX
  ☆22Updated last week
• MaximilianB2 / pc-gym
  Reinforcement learning environments for process control applications.
  ☆72Updated 2 months ago
• jeremiecoullon / jax-tqdm
  Add a tqdm progress bar to your JAX scans and loops.
  ☆124Updated 8 months ago
• resibots / pymap_elites
  map elites python reference implementation
  ☆96Updated 2 years ago
• TheodoreWolf / hyperoptax
  Parallel hyperparameter tuning with JAX
  ☆39Updated 6 months ago
• xiangyuan-zhang / controlgym
  Controlgym: Large-Scale Control Environments for Benchmarking Reinforcement Learning Algorithms
  ☆49Updated last year
• coax-dev / coax
  Modular framework for Reinforcement Learning in python
  ☆182Updated 2 years ago
• subho406 / pytorch2jax
  Pytorch2Jax is a small Python library that provides functions that wraps PyTorch models into Jax functions and Flax modules.
  ☆21Updated 2 years ago