Evolving Neural Networks in JAX by Robert Tjarko Lange

roberttlange.github.iosite

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

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   EvoJAX github.com

   tool category #jax link ↗

   Hardware-Accelerated Neuroevolution

   
2. ▲
   Getting started with JAX (MLPs, CNNs & RNNs) by Robert Lange roberttlange.github.io

   site category #jax link ↗

   Neural network building blocks from scratch with the basic JAX operators.

3. ▲
   JAX RL github.com

   tool category #jax link ↗

   Implementations of reinforcement learning algorithms.

   
4. ▲
   Meta-Learning in 50 Lines of JAX by Eric Jang blog.evjang.com

   site category #jax link ↗

   Introduction to both JAX and Meta-Learning.

5. ▲
   Introduction to JAX youtube.com

   video category #jax link ↗

   Simple neural network from scratch in JAX.

   
6. ▲
   Exploring hyperparameter meta-loss landscapes with JAX by Luke Metz lukemetz.com

   site category #jax link ↗

   Demonstrates how to use JAX to perform inner-loss optimization with SGD and Momentum, outer-loss optimization with gradients, and outer-loss optimization using evolutionary strategies.

JAX › Tutorials and Blog Posts: “Explores how JAX can power the next generation of scalable neuroevolution algorithms.”