qml.qaoa¶
This module provides a collection of methods that help in the construction of QAOA workflows.
We can demonstrate the PennyLane QAOA functionality with a basic application of QAOA: solving the MaxCut problem. We begin by defining the set of wires on which QAOA is executed, as well as the graph on which we will perform MaxCut. The node labels of the graph are the index of the wire to which they correspond:
import pennylane as qml
from pennylane import qaoa
from networkx import Graph
# Defines the wires and the graph on which MaxCut is being performed
wires = range(3)
graph = Graph([(0, 1), (1, 2), (2, 0)])
We now obtain the QAOA cost and mixer Hamiltonians for MaxCut on the graph that we defined:
# Defines the QAOA cost and mixer Hamiltonians
cost_h, mixer_h = qaoa.maxcut(graph)
These cost and mixer Hamiltonians are then used to define layers of the variational QAOA ansatz, which we implement as the following function:
# Defines a layer of the QAOA ansatz from the cost and mixer Hamiltonians
def qaoa_layer(gamma, alpha):
qaoa.cost_layer(gamma, cost_h)
qaoa.mixer_layer(alpha, mixer_h)
Finally, the full QAOA circuit is built. We begin by initializing the wires in an even superposition over
computational basis states, and then repeatedly apply QAOA layers with the
qml.layer
method. In this case we repeat the circuit twice:
# Repeatedly applies layers of the QAOA ansatz
def circuit(params, **kwargs):
for w in wires:
qml.Hadamard(wires=w)
qml.layer(qaoa_layer, 2, params[0], params[1])
With the circuit defined, we call the device on which QAOA will be executed, as well as the qml.VQECost
, which
creates the QAOA cost function: the expected value of the cost Hamiltonian with respect to the parametrized output
of the QAOA circuit.
# Defines the device and the QAOA cost function
dev = qml.device('default.qubit', wires=len(wires))
cost_function = qml.VQECost(circuit, cost_h, dev)
>>> print(cost_function([[1, 1], [1, 1]]))
1.8260274380964299
The QAOA cost function can then be optimized in the usual way, by calling one of the builtin PennyLane optimizers and updating the variational parameters until the expected value of the cost Hamiltonian is minimized.
Mixer Hamiltonians¶
Methods for constructing QAOA mixer Hamiltonians.
Cost Hamiltonians¶
Methods for generating QAOA cost Hamiltonians corresponding to different optimization problems.
QAOA Layers¶
Methods that define cost and mixer layers for use in QAOA workflows.
Functions¶

Applies the QAOA cost layer corresponding to a cost Hamiltonian. 

Applies the QAOA mixer layer corresponding to a mixer Hamiltonian. 
Contents
Downloads