Closed alantian closed 2 years ago
Experimental CMA-ES optimizer using JAX backend --- this is an adaption from https://github.com/CyberAgentAILab/cmaes/blob/main/cmaes/_cma.py which is a faithful implementation of CMA-ES described in https://arxiv.org/abs/1604.00772.
The adaption mainly replaces the Numpy backend with JAX one, and introduces several adjustments to facilitate efficient computation in JAX.
This is an experimental implementation for now and needs more revising.
LGTM
lerrytang
commented
2 years ago lerrytang
commented
2 years ago
Experimental CMA-ES optimizer using JAX backend --- this is an adaption from https://github.com/CyberAgentAILab/cmaes/blob/main/cmaes/_cma.py which is a faithful implementation of CMA-ES described in https://arxiv.org/abs/1604.00772.
The adaption mainly replaces the Numpy backend with JAX one, and introduces several adjustments to facilitate efficient computation in JAX.
This is an experimental implementation for now and needs more revising.