google / evojax

Related projects ⓘ

Alternatives and complementary repositories for evojax

• RobertTLange / evosax
  Evolution Strategies in JAX 🦎
  ☆506Updated 3 weeks ago
• google-deepmind / chex
  ☆788Updated this week
• RobertTLange / gymnax
  RL Environments in JAX 🌍
  ☆652Updated 4 months ago
• google-deepmind / rlax
  ☆1,263Updated last month
• nnaisense / evotorch
  Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.
  ☆1,016Updated this week
• instadeepai / jumanji
  🕹️ A diverse suite of scalable reinforcement learning environments in JAX
  ☆638Updated this week
• adaptive-intelligent-robotics / QDax
  Accelerated Quality-Diversity
  ☆266Updated last month
• google-deepmind / distrax
  ☆536Updated 2 months ago
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆1,699Updated last week
• luchris429 / purejaxrl
  Really Fast End-to-End Jax RL Implementations
  ☆724Updated 2 months ago
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆389Updated last week
• google / jaxopt
  Hardware accelerated, batchable and differentiable optimizers in JAX.
  ☆933Updated 2 months ago
• icaros-usc / pyribs
  A bare-bones Python library for quality diversity optimization.
  ☆211Updated 3 months ago
• 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
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆321Updated this week
• facebookresearch / moolib
  A library for distributed ML training with PyTorch
  ☆366Updated last year
• salesforce / warp-drive
  Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)
  ☆464Updated 3 months ago
• coax-dev / coax
  Modular framework for Reinforcement Learning in python
  ☆168Updated last year
• sotetsuk / pgx
  ♟️ Vectorized RL game environments in JAX
  ☆412Updated this week
• facebookresearch / salina
  a Lightweight library for sequential learning agents, including reinforcement learning
  ☆427Updated last year
• facebookresearch / minihack
  MiniHack the Planet: A Sandbox for Open-Ended Reinforcement Learning Research
  ☆478Updated 3 months ago
• n2cholas / awesome-jax
  JAX - A curated list of resources https://github.com/google/jax
  ☆1,562Updated 4 months ago
• google-deepmind / lab2d
  A customisable 2D platform for agent-based AI research
  ☆422Updated last year
• google-deepmind / launchpad
  ☆314Updated last year
• patrick-kidger / jaxtyping
  Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
  ☆1,219Updated this week
• google-deepmind / xmanager
  A platform for managing machine learning experiments
  ☆818Updated 3 months ago
• google / orbax
  Orbax provides common checkpointing and persistence utilities for JAX users
  ☆303Updated this week
• google-deepmind / reverb
  Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research
  ☆708Updated 2 weeks ago
• patrick-kidger / equinox
  Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
  ☆2,123Updated 3 weeks ago
• google-deepmind / meltingpot
  A suite of test scenarios for multi-agent reinforcement learning.
  ☆621Updated 2 weeks ago