Notice

This part is based on the JAX-Version EvoX.

To run this example, please install EvoX with the jax-version branch: (v0.9.1-dev)

Use this command to install EvoX's JAX Version:

pip install git+https://github.com/EMI-Group/evox/tree/v0.9.1-dev