Merge branch 'autograd-use-jpc' of https://github.com/PennyLaneAI/pen…

…nylane into autograd-use-jpc
PennyLaneAI · Oct 17, 2023 · 8d4fc52 · 8d4fc52
2 parents 2ac753b + f565bc3
commit 8d4fc52
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/pennylane/devices/execution_config.py b/pennylane/devices/execution_config.py
@@ -57,9 +57,10 @@ class ExecutionConfig:
     use_device_jacobian_product: Optional[bool] = None
     """Whether or not to use the device provided vjp or jvp to compute gradients.
 
-    ``None`` indicates to use the device if possible, but to fall back to device jacobian or pennylane behavior if it isn't.
+    ``None`` indicates to use the device if possible, but to fall back to the device Jacobian
+    or PennyLane behavior if it isn't.
 
-    True indicates a request to either use the device jacobian products fail.
+    ``True`` indicates to either use the device Jacobian products or fail.
     """
 
     gradient_method: Optional[str] = None

diff --git a/pennylane/interfaces/autograd.py b/pennylane/interfaces/autograd.py
@@ -80,7 +80,9 @@ def autograd_execute(
     Args:
         tapes (Sequence[.QuantumTape]): batch of tapes to execute
         execute_fn (Callable[[Sequence[.QuantumTape]], ResultBatch]): a function that turns a batch of circuits into results
-        jpc (JacobianProductCalculator): a class that can compute the vector jacobian product for the input tapes.
+        jpc (JacobianProductCalculator): a class that can compute the vector Jacobian product (VJP)
+            for the input tapes.
+
 
     Returns:
         TensorLike: A nested tuple of tape results. Each element in
@@ -117,7 +119,9 @@ def _execute(
             This argument should be generated from the provided list of tapes.
         tapes (Sequence[.QuantumTape]): batch of tapes to execute
         execute_fn (Callable[[Sequence[.QuantumTape]], ResultBatch]): a function that turns a batch of circuits into results
-        jpc (JacobianProductCalculator): a class that can compute the vector jacobian product for the input tapes.
+        jpc (JacobianProductCalculator): a class that can compute the vector Jacobian product (VJP)
+            for the input tapes.
+
     """
     return execute_fn(tapes)
 
@@ -138,7 +142,9 @@ def vjp(
             This argument should be generated from the provided list of tapes.
         tapes (Sequence[.QuantumTape]): batch of tapes to execute
         execute_fn (Callable[[Sequence[.QuantumTape]], ResultBatch]): a function that turns a batch of circuits into results
-        jpc (JacobianProductCalculator): a class that can compute the vector jacobian product for the input tapes.
+        jpc (JacobianProductCalculator): a class that can compute the vector Jacobian product (VJP)
+            for the input tapes.
+
 
     Returns:
         function: this function accepts the backpropagation