The examples provided in this repository are for experimental and educational purposes only. They demonstrate concepts and techniques but are not intended for direct use in production environments.

Then install JAX via the instructions here. The algorithms should be run on GPU to achieve efficiency gains. However, the code is supported on CPU for testing and investigation. We have tested the ...