The Example files contain all the code necessary to reproduce the Figures

Acknowledgement

The complex pytorch library comes from Qucumber https://github.com/PIQuIL/QuCumber