I would be interested in adding support for [neuron](https://awsdocs-neuron.readthedocs-hosted.com/en/latest/frameworks/jax/index.html) and trying out the solver on AWS hardware.
I would be interested in adding support for neuron and trying out the solver on AWS hardware.