Implementing evolutionary optimization for neural networks in JAX

[ttt733]

I've been using JAX for a little while now on personal projects, and I only understand it as well as I do because of some of the very helpful articles the community has put together. I decided to write up a post explaining some of the stuff I've been doing in my little niche. If OpenAI's post on evolution strategies ever got you curious, you can check here to see how I implemented something similar in JAX.

https://github.com/ttt733/nn-evolution-jax/blob/main/tutorial.md

I'm nowhere near a master of JAX, so if there's anything I got wrong or could do better, feel free to let me know!