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@v0.9.1-dev"