qml.capture.primitives.while_loop_prim

while_loop_prim = while_loop

A subclass to JAX’s Primitive that works like a Python function when evaluating JVPTracers and BatchTracers.

code/api/pennylane.capture.primitives.while_loop_prim

 

Download Python script

 

Download Notebook

 

View on GitHub