JAX with default.mixed device

Hey @Javier!

if I want to run a quantum circuit with this device using the GPU, do I have to change something in my code?

Unfortunately default.mixed is not currently compatible with running on GPU. It works using an all NumPy-based implementation. Though, as @antalszava pointed out, it’s great to have this as a feature request!

The idea would be to modify the circuit introducing some parameters and then do some gradient descent rounds, but for now I want to keep it simple.

That should still be possible with default.mixed, let us know if you need a hand with that part!

1 Like