EvoJax: Hardware Accelerated NeuroEvolution
Above is a short presentation on the EvoJAX hardware accelerated neuroevolution framework presented at GECCO 2022. The associated EvoJAX paper is available here. Another paper that uses EvoJAX is Modern Evolution Strategies for Creativity: Fitting Concrete Images and Abstract Concepts (NeurIPS Creativity Workshop 2021, EvoMUSART 2022)EvoJAX is a scalable, general purpose, hardware-accelerated neuroevolution toolkit. Built on top of the JAX library, this toolkit enables neuroevolution algorithms to work with neural networks running in parallel across multiple TPU/GPUs. EvoJAX achieves very high performance by implementing the evolution algorithm, neural network and task all in NumPy, which is compiled just-in-time to run on accelerators.