This repository contains Jupyter notebooks and supporting files for quantum computing training using CUDA-Q. These training materials have been developed by NVIDIA Corporation and are provided free of charge. Please see LICENSE for license details.
Instructions to install CUDA-Q can be found in the instructions.md file. If you do not have a local installation of CUDA-Q running on a GPU, the notebooks can be opened in qBraid Lab or in Google Colab. Simply click on the Launch on qBraid icon below or navigate to the notebook in github and select the Go to Colab icon at the top of the page. Note that using Google Colab will require additional steps outlined in the notebooks to install CUDA-Q.
If using qBraid Lab, use the Environment Manager to install the CUDA-Q environment and then activate it in your notebook. In qBraid Lab you can switch to a GPU instance using the Compute Manager.
Currently this repository contains one module: QAOA for Max Cut. More folders will be added as material becomes available.
The folder titled qaoa-for-max-cut
contain the Divide-and-Conquer QAOA for Max Cut module.
This material, which includes Jupyter notebooks and Python scripts, is organized into labs that build upon one another. The goal of the labs is to apply a divide-and-conquer QAOA algorithm to a large max cut problem using parallel computation. Lab 0 gives an overview of the learning material and an introduction to working with the Jupyter notebooks in Google CoLaboratory. Labs 1, 2, and 3 provide instructional material including solutions to exercises, while Lab 4 can serve as an open-ended assessment.
The Max Cut problem is an optimization problem defined as: Given a graph G, find the maximum cut of G, where the maximum cut (max cut) of a graph is defined to be a partitioning of the vertices into two disjoint sets so that the number of edges between the two partitions is maximized. The Max Cut problem is a NP-hard problem, and there is a rich body of research to develop classical and quantum algorithms to solve and/or approximate the max cut for large subclasses of graphs. Some of these algorithms fall under the divide-and-conquer paradigm. Divide and conquer breaks a large problem into smaller problems which are simple enough to be solved directly. Additionally, this paradigm lends itself to parallel computation since the smaller problems can often be solved independently. Recently, the divide-and-conquer paradigm has been applied to the Quantum Approximation Optimization Algorithm (QAOA) for max cut (arXiv:2406.17383, arXiv:2205.11762v1, arxiv.2101.07813v1, arxiv:2304.03037v1, and arxiv:2009.06726). In this tutorial, we will introduce this algorithm and implement parallel computation with CUDA-Q.
This tutorial begins in Lab 1 with a demonstration of solving the Max Cut problem for a small graph with QAOA. To set the groundwork for the remaining labs, Lab 1 ends with a preview of the divide-and-conquer paradigm. In Lab 2, we walk through one level of the divide-and-conquer algorithm, and we follow that up in Lab 3 with the recursive implementation for much larger and denser graphs. Additionally in Labs 2 and 3, we experiment with running quantum circuits in parallel on GPUs. Finally, learners can take the assessment in Lab 4 in which they are challenged to approximate the weighted max cut problem.
The learning objectives of this tutorial are:
- Execute the QAOA algorithm to find approximate max cuts of a given graph using CUDA-Q
- Understand the limitations of the QAOA algorithm for solving max cut in the NISQ era
- Apply QAOA to find an approximate solution to Quadratic Unconstrained Binary Optimization problems
- Make various adjustments to the QAOA algorithm to improve results
- Simulate quantum circuits in parallel on multiple GPUs to speed up overall run time using CUDA-Q
Pre-requisites:
- Familiarity with Python with enough comfort to refer to Python package documentation, specifically NetworkX, as needed
- Familiarity with variational quantum algorithms (e.g. VQE or QAOA)
To learn more about CUDA-Q, including additional examples and tutorials, visit the NVIDIA CUDA-Q documentation page.