Reinforcement Learning for Quantum Variational Circuit Optimization

Code for Optimizing Quantum Variational Circuits with Deep Reinforcement Learning

To use the pretrained models, don't worry about training or testing. Download deploy folder and use the mixed function in augment.py to work with your circuit. See the file for the required inputs. Note that deployment usage requires stable_baselines3 (which requires PyTorch) and numpy.

To recreate the results from the paper, use the testing folder. Each file runs and outputs the information as presented in the table. Note that this has dependencies on TensorFlow, TensorFlow-Quantum, sklearn, and stable_baselines3 (which requires PyTorch).

To train your own agent, go to the training folder. Specifiy the maximum sizes and training durations for the agent and run the code.

Examples for Deployment

The example (written in pennylane) is similar to that used to generate the barren plateaus figures in the paper.

Questions?

Join the rl_opt channel in the Unitary Fund discord.