Can FMMAX be mounted to docker?

[Simulator-CEM]

I hope to try the FMMAX+GPU AD optimization, but it is only available in Linux/Unix or Win WL2, which I am not good at. I have tried several codes for Linux system via the docker platform, such as the EMUStack and the gyptis. So I wonder if FMMAX can work in a similar way.

Besides, it seem the JAX.eig() can work with GPU via "magma", but does it really accelerate the algorithm? Thank you so much.

[mfschubert]

Hi @Simulator-CEM , I think this is more of a question about jax. If you can get jax running, then fmmax should also run. Perhaps check out this resource. https://docs.jax.dev/en/latest/installation.html#nvidia-gpu-docker-containers