Simple implementations of Cartesian Genetic Programming (CGP) and Linear Genetic Programming (LGP) in JAX. The main goal is to interface CGP and LGP with Brax (an implementation of several Mujoco environments in JAX) to leverage parallelization on GPU and speed up the optimization.