Uso de la compilación paramétrica para acelerar las tareas híbridas

Amazon Braket admite la compilación paramétrica en algunos casos. QPUs Esto le permite reducir la sobrecarga asociada al costoso paso de compilación desde el punto de vista computacional al compilar un circuito solo una vez y no para cada iteración de su algoritmo híbrido. Esto puede mejorar considerablemente los tiempos de ejecución de Hybrid Jobs, ya que evita la necesidad de volver a compilar el circuito en cada paso. Simplemente envíe los circuitos parametrizados a uno de nuestros trabajos compatibles QPUs como Braket Hybrid. Para trabajos híbridos de larga duración, Braket utiliza automáticamente los datos de calibración actualizados del proveedor del hardware al compilar el circuito para garantizar resultados de la más alta calidad.

Para crear un circuito paramétrico, primero debe proporcionar los parámetros como entradas en el script de su algoritmo. En este ejemplo, utilizamos un circuito paramétrico pequeño e ignoramos cualquier procesamiento clásico entre cada iteración. Para las cargas de trabajo típicas, debe enviar muchos circuitos por lotes y realizar el procesamiento clásico, como la actualización de los parámetros en cada iteración.

import os

from braket.aws import AwsDevice
from braket.circuits import Circuit, FreeParameter

def start_here():

    print("Test job started.")

    # Use the device declared in the job script
    device = AwsDevice(os.environ["AMZN_BRAKET_DEVICE_ARN"])

    circuit = Circuit().rx(0, FreeParameter("theta"))
    parameter_list = [0.1, 0.2, 0.3]
    
    for parameter in parameter_list:
        result = device.run(circuit, shots=1000, inputs={"theta": parameter})

    print("Test job completed.")

Puede enviar el script del algoritmo para que se ejecute como un Hybrid Job con el siguiente script de trabajo. Cuando se ejecuta el Hybrid Job en un circuito QPU que admite la compilación paramétrica, el circuito se compila solo en la primera ejecución. En las siguientes ejecuciones, el circuito compilado se reutiliza, lo que aumenta el rendimiento en tiempo de ejecución del Hybrid Job sin líneas de código adicionales.

from braket.aws import AwsQuantumJob

job = AwsQuantumJob.create(
    device=device_arn,
    source_module="algorithm_script.py",
)