jaunatisblue/final-qibotn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixing grad nan

Browse Source
This commit is contained in:
Mattia Robbiano
2025-09-20 15:15:40 +02:00
parent 63266c5ba0
commit 9853e86deb
2 changed files with 70 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/quimb_intro/test.py
View File

@@ -30,13 +30,14 @@ def build_circuit(nqubits, nlayers):
for q in range(nqubits):
circ.add(gates.RY(q=q, theta=0.))
circ.add(gates.RZ(q=q, theta=0.))
[circ.add(gates.CNOT(q%nqubits, (q+1)%nqubits) for q in range(nqubits))]
[circ.add(gates.CZ(q % nqubits, (q + 1) % nqubits)) for q in range(nqubits)]
circ.add(gates.M(*range(nqubits)))
return circ
nqubits = 4
circuit = build_circuit(nqubits=nqubits, nlayers=3)
quimb_circuit = quimb_backend._qibo_circuit_to_quimb(circuit)
def f(params):
circuit.set_parameters(params)
@@ -46,5 +47,4 @@ def f(params):
)
parameters = np.random.uniform(-np.pi, np.pi, size=len(circuit.get_parameters()))
print(f(parameters))
print(jax.value_and_grad(f)(parameters))
print(jax.value_and_grad(f)(parameters))