qflux 0.0.3__py3-none-any.whl → 0.0.4__py3-none-any.whl

qflux/GQME/dynamics_GQME.py

@@ -59,7 +59,7 @@ class DynamicsGQME:
 
     
     def get_Liouvillian(self) -> None:
-        """
+        r"""
         Construct the Liouvillian superoperator using the system Hamiltonian.
         
         The Liouvillian is defined as:

qflux/closed_systems/__init__.py

@@ -15,3 +15,4 @@ spin_dynamics_oo.py  --> SpinDynamicsS, SpinDynamicsH : Statevector and hadamard
 from .classical_methods import DynamicsCS
 from .qubit_methods import QubitDynamicsCS
 from .spin_dynamics_oo import SpinDynamicsS, SpinDynamicsH
+from .direct_method import hamiltonian_simulation

qflux/closed_systems/direct_method.py

@@ -0,0 +1,119 @@
+import numpy as np
+from qiskit import QuantumCircuit, QuantumRegister
+
+def exp_all_z(circuit, quantum_register, pauli_indexes, control_qubit=None, t=1):
+    r"""
+    Implements \( e^{-i t Z \otimes \cdots \otimes Z} \) on specified qubits.
+    
+    Args:
+        circuit (QuantumCircuit): The circuit to modify.
+        quantum_register (QuantumRegister): Register containing target qubits.
+        pauli_indexes (list): Indices of qubits where \( Z \) acts.
+        control_qubit (Qubit, optional): Optional control qubit for conditional application.
+        t (float): Evolution time.
+    
+    Returns:
+        QuantumCircuit: Updated circuit with the operation applied.
+    """
+    if control_qubit and control_qubit.register not in circuit.qregs:
+        circuit.add_register(control_qubit.register)
+
+    if not pauli_indexes:
+        if control_qubit:
+            circuit.p(t, control_qubit)  # Phase gate
+        return circuit
+
+    # Parity computation
+    for i in range(len(pauli_indexes) - 1):
+        circuit.cx(quantum_register[pauli_indexes[i]], quantum_register[pauli_indexes[i + 1]])
+
+    # Apply phase rotation
+    target = quantum_register[pauli_indexes[-1]]
+    angle = -2 * t
+    if control_qubit:
+        circuit.crz(angle, control_qubit, target)
+    else:
+        circuit.rz(angle, target)
+
+    # Uncompute parity
+    for i in reversed(range(len(pauli_indexes) - 1)):
+        circuit.cx(quantum_register[pauli_indexes[i]], quantum_register[pauli_indexes[i + 1]])
+
+    return circuit
+
+
+def exp_pauli(pauli, quantum_register, control_qubit=None, t=1):
+    r"""
+    Implements \( e^{-i t P} \) for a Pauli string \( P \).
+
+    Args:
+        pauli (str): Pauli string (e.g., "XIZY").
+        quantum_register (QuantumRegister): Target register.
+        control_qubit (Qubit, optional): Optional control qubit.
+        t (float): Evolution time.
+
+    Returns:
+        QuantumCircuit: Circuit implementing the Pauli evolution.
+    """
+    if len(pauli) != len(quantum_register):
+        raise ValueError("Pauli string length must match register size.")
+
+    pauli_indexes = []
+    pre_circuit = QuantumCircuit(quantum_register)
+
+    for i, op in enumerate(pauli):
+        if op == 'I':
+            continue
+        elif op == 'X':
+            pre_circuit.h(i)
+            pauli_indexes.append(i)
+        elif op == 'Y':
+            pre_circuit.rx(np.pi/2, i)
+            pauli_indexes.append(i)
+        elif op == 'Z':
+            pauli_indexes.append(i)
+        else:
+            raise ValueError(f"Invalid Pauli operator '{op}' at position {i}.")
+
+    circuit = QuantumCircuit(quantum_register)
+    circuit.compose(pre_circuit, inplace=True)
+    circuit = exp_all_z(circuit, quantum_register, pauli_indexes, control_qubit, t)
+    circuit.compose(pre_circuit.inverse(), inplace=True)
+    return circuit
+
+
+def hamiltonian_simulation(hamiltonian, quantum_register=None, control_qubit=None, t=1, trotter_number=1):
+    r"""
+    Implements \( e^{-i H t} \) using first-order Trotterization.
+    
+    Args:
+        hamiltonian (dict): Pauli terms with coefficients (e.g., {"ZZ": 0.5, "XX": 0.3}).
+        quantum_register (QuantumRegister, optional): Target register.
+        control_qubit (Qubit, optional): Optional control qubit.
+        t (float): Simulation time.
+        trotter_number (int): Number of Trotter steps.
+    
+    Returns:
+        QuantumCircuit: Trotterized Hamiltonian evolution circuit.
+    """
+    if not hamiltonian:
+        raise ValueError("Hamiltonian must contain at least one term.")
+    
+    n_qubits = len(next(iter(hamiltonian)))
+    if quantum_register is None:
+        quantum_register = QuantumRegister(n_qubits)
+    
+    delta_t = t / trotter_number
+    circuit = QuantumCircuit(quantum_register)
+    
+    for pauli_str, coeff in hamiltonian.items():
+        term_circuit = exp_pauli(pauli_str, quantum_register, control_qubit, coeff * delta_t)
+        circuit.compose(term_circuit, inplace=True)
+    
+    full_circuit = QuantumCircuit(quantum_register)
+    for _ in range(trotter_number):
+        full_circuit.compose(circuit, inplace=True)
+    
+    return full_circuit
+
+

qflux/closed_systems/spin_dynamics_oo.py

@@ -4,6 +4,9 @@ from .utils import execute
 from qiskit import QuantumCircuit, QuantumRegister, ClassicalRegister
 from qiskit_aer import Aer
 from .spin_propagators import get_time_evolution_operator
+import numpy.typing as npt
+import os 
+
 
 class SpinDynamicsS:
     """
@@ -37,7 +40,6 @@ class SpinDynamicsS:
             evolution_timestep (float): Time step for the evolution.
             trotter_steps (int): Number of Trotter steps for the simulation.
             hamiltonian_coefficients (list): Hamiltonian coefficients for the system.
-            initial_state (str or list): Initial state of the system, represented as a binary string or list.
         """
         self.num_qubits = num_qubits
         self.evolution_timestep = evolution_timestep
@@ -64,7 +66,7 @@ class SpinDynamicsS:
         Prepare the initial state vector from the binary string or list.
 
         Returns:
-            ndarray: Flattened initial state vector.
+            psin (npt.ArrayLike): Flattened initial state vector.
         """
         zero_state = np.array([[1], [0]])
         one_state = np.array([[0], [1]])
@@ -83,12 +85,13 @@ class SpinDynamicsS:
     def qsolve_statevector(self, psin):
         """
         Perform statevector propagation for the quantum circuit.
+            initial_state (str or list): Initial state of the system, represented as a binary string or list.
 
         Args:
-            psin (ndarray): Input statevector.
+            psin (npt.ArrayLike): Input statevector.
 
         Returns:
-            ndarray: Final statevector after execution.
+            ndarray (npt.ArrayLike): Final statevector after execution.
         """
         d = int(np.log2(np.size(psin)))
         qre = QuantumRegister(d)
@@ -204,7 +207,7 @@ class SpinDynamicsH:
             imag_expectation (bool): Whether to evaluate the imaginary component (default: False).
 
         Returns:
-            QuantumCircuit: The constructed Hadamard test circuit.
+            qc (QuantumCircuit): The constructed Hadamard test circuit.
         """
         qr = QuantumRegister(self.num_qubits + 1)
         cr = ClassicalRegister(1)
@@ -233,7 +236,7 @@ class SpinDynamicsH:
             num_shots (int): Number of shots for circuit execution (default: 100).
 
         Returns:
-            dict: Measurement counts.
+            dict (dict): Measurement counts.
         """
         job = execute(qc, self.simulator, shots=num_shots)
         return job.result().get_counts()
@@ -247,7 +250,7 @@ class SpinDynamicsH:
             counts (dict): Measurement counts.
 
         Returns:
-            float: Average spin correlation.
+            results (float): Average spin correlation.
         """
         qubit_to_spin_map = {'0': 1, '1': -1}
         total_counts = sum(counts.values())
@@ -264,7 +267,7 @@ class SpinDynamicsH:
             state_string (str): Binary string representing the initial state.
 
         Returns:
-            QuantumCircuit: Circuit for initializing the state.
+            qc (QuantumCircuit): Circuit for initializing the state.
         """
         state_string = ''.join('1' if char == '0' else '0' for char in state_string)
         qr = QuantumRegister(num_qubits)
@@ -325,10 +328,14 @@ class SpinDynamicsH:
             prefix (str): Prefix for loading precomputed reference data.
         """
         abs_corr = np.abs(np.array(self.real_amp_list) + 1j * np.array(self.imag_amp_list))
-
+        
         plt.plot(self.time_range, abs_corr, '.', label='Hadamard Test')
-        ref_sa = np.load(f'../../../data/{self.num_qubits}_spin_chain_SA_obs.npy')
-        ref_time = np.load(f'../../../data/{self.num_qubits}_spin_chain_time.npy')
+        reference_filename = f'data/{self.num_qubits}_spin_chain_SA_obs.npy'
+        if os.path.exists(reference_filename):
+            ref_sa = np.load(f'data/{self.num_qubits}_spin_chain_SA_obs.npy')
+            ref_time = np.load(f'data/{self.num_qubits}_spin_chain_time.npy')
+        else:
+            print('No reference data file found!')
         plt.plot(ref_time, ref_sa, '-', label='Statevector')
 
         plt.xlabel('Time')

{qflux-0.0.3.dist-info → qflux-0.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qflux
-Version: 0.0.3
+Version: 0.0.4
 Summary: qflux is a package for running quantum dynamics calculations on quantum devices.
 Author-email: Brandon Allen <brandon.allen@yale.edu>, Delmar Cabral <delmar.azevedocabral@yale.edu>, Alexander Soudackov <alexander.soudackov@yale.edu>, Anton Morgunov <anton@ischemist.com>
 License: MIT

{qflux-0.0.3.dist-info → qflux-0.0.4.dist-info}/RECORD

@@ -1,17 +1,18 @@
 qflux/__init__.py,sha256=j0hovUC2F2GdGooe2jExe89bXQh4HSozB26mOmOx03o,129
 qflux/GQME/__init__.py,sha256=zeTbgWoan3pUfp5aa7MU-ywWBTHAOp4RpLNE4IXW6bU,115
-qflux/GQME/dynamics_GQME.py,sha256=pFK75IqyTsLhODKSHnQiRMyQn8YDgyt-eFPiFAHjwUQ,16849
+qflux/GQME/dynamics_GQME.py,sha256=mi9UUlKsks8dYRHEenESxfmiUX5WYQksFn4E-F_uP_Y,16850
 qflux/GQME/params.py,sha256=u7mHRdbbf88ccTIvw9Hb4AunlfNBrzara0OZTL80vwk,1822
 qflux/GQME/readwrite.py,sha256=JUUEsDL8Ep705P8iwlQ4YjAm7MswQ-eNpH1wa8_P-_I,3583
 qflux/GQME/tdvp.py,sha256=32Z_cLJornzM4D63nmqekhwtc-OosrQcZhIeEZAqq9s,7136
 qflux/GQME/tt_tfd.py,sha256=-RPH9aiCinLzG0VdZo43UBVKWcuoplkFmi77oOtv5zU,15204
 qflux/closed_systems/VarQTE.py,sha256=rNDFnRmF_jDv-wSSaBnFp0ZwrgjOJ3NdqyoT4WUlX4E,17840
-qflux/closed_systems/__init__.py,sha256=CBnnqCBp0MD7jCGq1P0nzwCyR3okp4jkC8tvqWUTBCY,472
+qflux/closed_systems/__init__.py,sha256=WUo9qGoEV3BX-RrsLUjW70COREXmQydvLc-M1S_kSHs,522
 qflux/closed_systems/classical_methods.py,sha256=Etqjd9OkqCgPzhC2-FgQWlIiTu66V2_7LRftPDwBrb8,15882
 qflux/closed_systems/custom_execute.py,sha256=Mk24DEvEbQlnu52MqhnSOGBpplHoAMz5__AcBKo1E_4,656
+qflux/closed_systems/direct_method.py,sha256=d1s7cF4J3BauaQhGUXqbZLu81EsuZqMlYz8UJCZX0v4,4120
 qflux/closed_systems/hamiltonians.py,sha256=ArdljgB7ASUq5QIKdN8CRdhf7b_2nFyb8OPacOGGvEs,3156
 qflux/closed_systems/qubit_methods.py,sha256=2TFeR7lxAqAu3CYUPQqUJtN9bMPpKTukHgGohiJJASk,10883
-qflux/closed_systems/spin_dynamics_oo.py,sha256=cn4aRM3wYkP6CBbL-LhlUMR4dqQSQK7bpSV-twSzFbU,13925
+qflux/closed_systems/spin_dynamics_oo.py,sha256=dmtT1MdiKHlYuPU56B6Lndfsyv7rAlIYfm6kritZtDI,14213
 qflux/closed_systems/spin_propagators.py,sha256=pMg4ttThkXw2kzBlY2yDpdomTn-7alRtpKUAxD-vSl8,11496
 qflux/closed_systems/utils.py,sha256=eiB5V0xvi43-jC35nKl2kMY_vr9r_-2njNn5qPLh494,6975
 qflux/open_systems/__init__.py,sha256=pHbE5rHa0ktH9ZCWUECbmgXQxeS_dJ4PQ5GBhtM9uJk,102
@@ -32,8 +33,8 @@ qflux/variational_methods/qmad/ansatz.py,sha256=2ttVOrueLYycyXaBy1uxPkaTNEtkctL9
 qflux/variational_methods/qmad/ansatzVect.py,sha256=OJdRDhKBYou93fwetri5KXFoAj9N4cdjj799Nqu-qmE,2934
 qflux/variational_methods/qmad/effh.py,sha256=l74femDGDDVCiHerfDDj2k30phyrYIlI_ssn4YU426g,3722
 qflux/variational_methods/qmad/solver.py,sha256=1B9DpcuT7bpKcIWt1kHkxT3-gwQECdEzKFcBtJlFYDg,12751
-qflux-0.0.3.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-qflux-0.0.3.dist-info/METADATA,sha256=sZdUGSirgVQ6W6yT5E-jZp4BTVCltcDlme9AJ6ILGx4,6871
-qflux-0.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-qflux-0.0.3.dist-info/top_level.txt,sha256=GZgw1Z1DZDPg78NCF1dXCHw_f4usRh5HP5yLtgECIpo,6
-qflux-0.0.3.dist-info/RECORD,,
+qflux-0.0.4.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+qflux-0.0.4.dist-info/METADATA,sha256=FRoWZNZ_v5o80Ja7YINnLTpJnkDPJ0TITuNb5lPnIZM,6871
+qflux-0.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+qflux-0.0.4.dist-info/top_level.txt,sha256=GZgw1Z1DZDPg78NCF1dXCHw_f4usRh5HP5yLtgECIpo,6
+qflux-0.0.4.dist-info/RECORD,,