@qnode(dev, interface='jax', diff_method=diff_method)
def circuit(weights):
    op1 = qml.RY(weights[0], wires=0)
    op2 = qml.RX(weights[1], wires=0)
    return qml.expval(qml.PauliZ(wires=0))
@qnode(dev, interface='jax', diff_method=diff_method)
def circuit(weights):
    op1 = qml.RY(weights[0], wires=0)
    op2 = qml.RX(weights[1], wires=0)
    return qml.expval(qml.PauliZ(wires=0))