def circuit(x, y, input_state=np.array([0, 0])):
    qml.BasisState(input_state, wires=[0, 1])
    qml.RX(x, wires=[0])
    qml.RY(y, wires=[0])
    return qml.expval(qml.PauliZ(0))
def circuit(x, y, input_state=np.array([0, 0])):
    qml.BasisState(input_state, wires=[0, 1])
    qml.RX(x, wires=[0])
    qml.RY(y, wires=[0])
    return qml.expval(qml.PauliZ(0))