How to use PyTorch+Pennylane to implement this code?

PennyLane Help
1

Dropout for Quantum Neural Networks

This code is implemented using JAX, and I am curious about how to combine it with PyTorch to train with classical networks

2

Hi @HotFrog,

We have a lot of demos on using PyTorch and PennyLane which you can find here. However if your code is already in Jax you might find it easier to keep it in Jax and use Jax with PennyLane instead (you can find some demos on this here).

Keeping your code in Jax probably has more advantages than moving it to PyTorch. Let me know if you have any additional questions. I hope this helps!

3

Dropout for Quantum Neural Networks | PennyLane Demos
i am wondering if it can be moved into pytorch

4

Hi @HotFrog , parts of it might but you might have to remove things like jitting. You can try it and let us know how it goes.