EvoJAX

github.comtool

Hardware-Accelerated Neuroevolution

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   Evolving Neural Networks in JAX by Robert Tjarko Lange roberttlange.github.io

   site category #jax link ↗

   Explores how JAX can power the next generation of scalable neuroevolution algorithms.

2. ▲
   evosax github.com

   tool category #jax link ↗

   JAX-Based Evolution Strategies

   
3. ▲
   JAXopt github.com

   tool category #jax link ↗

   Hardware accelerated (GPU/TPU), batchable and differentiable optimizers in JAX.

   
4. ▲
   FedJAX github.com

   tool category #jax link ↗

   Federated learning in JAX, built on Optax and Haiku.

   
5. ▲
   mpi4jax github.com

   tool category #jax link ↗

   Combine MPI operations with your Jax code on CPUs and GPUs.

   
6. ▲
   Pax github.com

   tool category #jax link ↗

   A Jax-based machine learning framework for training large scale models.

JAX › Libraries > New Libraries: “Hardware-Accelerated Neuroevolution”