RobertTLange / evosax

Related projects ⓘ

Alternatives and complementary repositories for evosax

• RobertTLange / gymnax
  RL Environments in JAX 🌍
  ☆652Updated 4 months ago
• google / evojax
  ☆845Updated 4 months ago
• adaptive-intelligent-robotics / QDax
  Accelerated Quality-Diversity
  ☆266Updated last month
• instadeepai / jumanji
  🕹️ A diverse suite of scalable reinforcement learning environments in JAX
  ☆638Updated this week
• google-deepmind / chex
  ☆788Updated this week
• instadeepai / flashbax
  ⚡ Flashbax: Accelerated Replay Buffers in JAX
  ☆213Updated 3 weeks ago
• google-deepmind / distrax
  ☆536Updated 2 months ago
• icaros-usc / pyribs
  A bare-bones Python library for quality diversity optimization.
  ☆211Updated 3 months ago
• EdanToledo / Stoix
  🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
  ☆238Updated last week
• coax-dev / coax
  Modular framework for Reinforcement Learning in python
  ☆168Updated last year
• luchris429 / purejaxrl
  Really Fast End-to-End Jax RL Implementations
  ☆724Updated 2 months ago
• google / jaxopt
  Hardware accelerated, batchable and differentiable optimizers in JAX.
  ☆933Updated 2 months ago
• nnaisense / evotorch
  Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.
  ☆1,016Updated this week
• google-deepmind / rlax
  ☆1,263Updated last month
• google-research / rliable
  [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.
  ☆772Updated 3 months ago
• MichaelTMatthews / Craftax
  (Crafter + NetHack) in JAX. ICML 2024 Spotlight.
  ☆208Updated last month
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆332Updated this week
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆321Updated this week
• keraJLi / rejax
  ☆149Updated this week
• FLAIROx / JaxMARL
  Multi-Agent Reinforcement Learning with JAX
  ☆441Updated 2 weeks ago
• ikostrikov / jaxrl
  JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
  ☆631Updated 2 years ago
• sotetsuk / pgx
  ♟️ Vectorized RL game environments in JAX
  ☆412Updated this week
• araffin / sbx
  SBX: Stable Baselines Jax (SB3 + Jax)
  ☆343Updated this week
• EMI-Group / tensorneat
  GPU-accelerated NeuroEvolution of Augmenting Topologies (NEAT)
  ☆69Updated last week
• JaxGaussianProcesses / GPJax
  Gaussian processes in JAX.
  ☆461Updated 2 weeks ago
• wilson-labs / cola
  Compositional Linear Algebra
  ☆432Updated 3 weeks ago
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆389Updated last week
• epignatelli / navix
  Accelerated minigrid environments with JAX
  ☆119Updated 3 months ago
• ott-jax / ott
  Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
  ☆526Updated this week
• kenjyoung / MinAtar
  ☆286Updated last year