def testInversion(self):
'''
Try a simple three level SC qubit system and see if can prepare the excited state.
'''
#Setup a three level qubit and a 100MHz delta
Q1 = SCQubit(3, 4.987456e9, -100e6, name='Q1')
systemParams = SystemParams()
systemParams.add_sub_system(Q1)
systemParams.add_control_ham(
inphase=Hamiltonian(0.5 * (Q1.loweringOp + Q1.raisingOp)),
quadrature=Hamiltonian(0.5 *
(-1j * Q1.loweringOp + 1j * Q1.raisingOp)))
systemParams.create_full_Ham()
systemParams.measurement = Q1.levelProjector(1)
#Setup the pulse parameters for the optimization
pulseParams = PulseParams()
pulseParams.timeSteps = 1e-9 * np.ones(30)
pulseParams.rhoStart = Q1.levelProjector(0)
pulseParams.rhoGoal = Q1.levelProjector(1)
pulseParams.add_control_line(freq=-Q1.omega)
pulseParams.H_int = Hamiltonian(Q1.omega * np.diag(np.arange(Q1.dim)))
pulseParams.optimType = 'state2state'
#Call the optimization
optimize_pulse(pulseParams, systemParams)
#Now test the optimized pulse and make sure it puts all the population in the excited state
result = simulate_sequence(pulseParams,
systemParams,
pulseParams.rhoStart,
simType='unitary')[0]
assert result > 0.99
def setUp(self):
#Setup a simple non-coupled two qubit system
self.systemParams = SystemParams()
self.Q1 = SCQubit(2, 5e9, name='Q1')
self.systemParams.add_sub_system(self.Q1)
self.Q2 = SCQubit(2, 6e9, name='Q2')
self.systemParams.add_sub_system(self.Q2)
X = 0.5 * (self.Q1.loweringOp + self.Q1.raisingOp)
Y = 0.5 * (-1j * self.Q1.loweringOp + 1j * self.Q2.raisingOp)
self.systemParams.add_control_ham(
inphase=Hamiltonian(self.systemParams.expand_operator('Q1', X)),
quadrature=Hamiltonian(self.systemParams.expand_operator('Q1', Y)))
self.systemParams.add_control_ham(
inphase=Hamiltonian(self.systemParams.expand_operator('Q2', X)),
quadrature=Hamiltonian(self.systemParams.expand_operator('Q2', Y)))
self.systemParams.measurement = self.systemParams.expand_operator(
'Q1', self.Q1.pauliZ) + self.systemParams.expand_operator(
'Q2', self.Q2.pauliZ)
self.systemParams.create_full_Ham()
#Define Rabi frequency and pulse lengths
self.rabiFreq = 10e6
#Define the initial state as the ground state
self.rhoIn = np.zeros((self.systemParams.dim, self.systemParams.dim))
self.rhoIn[0, 0] = 1
def setUp(self):
#Setup the system
self.systemParams = SystemParams()
self.qubit = SCQubit(2, 0e9, name='Q1', T1=1e-6)
self.systemParams.add_sub_system(self.qubit)
#self.systemParams.add_control_ham(inphase = Hamiltonian(0.5*(self.qubit.loweringOp + self.qubit.raisingOp)), quadrature = Hamiltonian(0.5*(-1j*self.qubit.loweringOp + 1j*self.qubit.raisingOp)))
self.systemParams.add_control_ham(
inphase=Hamiltonian(0.5 * self.qubit.pauliX),
quadrature=Hamiltonian(0.5 * self.qubit.pauliY))
self.systemParams.measurement = self.qubit.pauliZ
self.systemParams.create_full_Ham()
#Define Rabi frequency and pulse lengths
self.rabiFreq = 10e6
self.pulseLengths = np.linspace(0, 100e-9, 40)
#Define the initial state as the ground state
self.rhoIn = self.qubit.levelProjector(0)
def setUp(self):
#Setup the system
self.systemParams = SystemParams()
self.qubit = SCQubit(3, 5e9, -100e6, name='Q1', T1=2e-6)
self.systemParams.add_sub_system(self.qubit)
self.systemParams.add_control_ham(
inphase=Hamiltonian(
0.5 * (self.qubit.loweringOp + self.qubit.raisingOp)),
quadrature=Hamiltonian(
-0.5 *
(-1j * self.qubit.loweringOp + 1j * self.qubit.raisingOp)))
self.systemParams.measurement = self.qubit.pauliZ
self.systemParams.create_full_Ham()
#Add the 2us T1 dissipator
self.systemParams.dissipators = [Dissipator(self.qubit.T1Dissipator)]
#Define the initial state as the ground state
self.rhoIn = self.qubit.levelProjector(0)
from copy import deepcopy
from scipy.constants import pi
from scipy.linalg import eigh
from PySim.SystemParams import SystemParams
from PySim.QuantumSystems import Hamiltonian, Dissipator
from PySim.PulseSequence import PulseSequence
from PySim.Simulation import simulate_sequence_stack, simulate_sequence
from PySim.QuantumSystems import SCQubit
from PySim.OptimalControl import optimize_pulse, PulseParams
#Setup the system
systemParams = SystemParams()
#First the two qubits defined in the lab frame
Q1 = SCQubit(numLevels=3, omega=4.76093e9, delta=-244e6, name='Q1', T1=10e-6)
systemParams.add_sub_system(Q1)
Q2 = SCQubit(numLevels=3, omega=5.34012e9, delta=-224e6, name='Q2', T1=10e-6)
systemParams.add_sub_system(Q2)
#Add an interaction between the qubits to get the the cross-resonance effect
systemParams.add_interaction('Q1', 'Q2', 'FlipFlop', 2e6)
#Create the full Hamiltonian
systemParams.create_full_Ham()
#Calculate the eigenframe for the natural Hamiltonian
def testDRAG(self):
'''
Try a unitary inversion pulse on a three level SCQuibt and see if we get something close to DRAG
'''
#Setup a three level qubit and a 100MHz delta
Q1 = SCQubit(3, 4.987456e9, -150e6, name='Q1')
systemParams = SystemParams()
systemParams.add_sub_system(Q1)
systemParams.add_control_ham(
inphase=Hamiltonian(0.5 * (Q1.loweringOp + Q1.raisingOp)),
quadrature=Hamiltonian(0.5 *
(-1j * Q1.loweringOp + 1j * Q1.raisingOp)))
systemParams.add_control_ham(
inphase=Hamiltonian(0.5 * (Q1.loweringOp + Q1.raisingOp)),
quadrature=Hamiltonian(0.5 *
(-1j * Q1.loweringOp + 1j * Q1.raisingOp)))
systemParams.create_full_Ham()
systemParams.measurement = Q1.levelProjector(1)
#Setup the pulse parameters for the optimization
numPoints = 30
pulseTime = 15e-9
pulseParams = PulseParams()
pulseParams.timeSteps = (pulseTime / numPoints) * np.ones(numPoints)
pulseParams.rhoStart = Q1.levelProjector(0)
pulseParams.rhoGoal = Q1.levelProjector(1)
pulseParams.Ugoal = Q1.pauliX
pulseParams.add_control_line(freq=-Q1.omega,
bandwidth=300e6,
maxAmp=200e6)
pulseParams.add_control_line(freq=-Q1.omega,
phase=-np.pi / 2,
bandwidth=300e6,
maxAmp=200e6)
pulseParams.H_int = Hamiltonian(
(Q1.omega) * np.diag(np.arange(Q1.dim)))
pulseParams.optimType = 'unitary'
pulseParams.derivType = 'finiteDiff'
#Start with a Gaussian
tmpGauss = np.exp(-np.linspace(-2, 2, numPoints)**2)
tmpScale = 0.5 / (np.sum(pulseParams.timeSteps * tmpGauss))
pulseParams.startControlAmps = np.vstack(
(tmpScale * tmpGauss, np.zeros(numPoints)))
#Call the optimization
optimize_pulse(pulseParams, systemParams)
if plotResults:
plt.plot(
np.cumsum(pulseParams.timeSteps) * 1e9,
pulseParams.controlAmps.T / 1e6)
plt.ylabel('Pulse Amplitude (MHz)')
plt.xlabel('Time (ns)')
plt.legend(('X Quadrature', 'Y Quadrature'))
plt.title('DRAG Pulse from Optimal Control')
plt.show()
#Now test the optimized pulse and make sure it does give us the desired unitary
result = simulate_sequence(pulseParams,
systemParams,
pulseParams.rhoStart,
simType='unitary')
assert np.abs(np.trace(np.dot(
result[1].conj().T, pulseParams.Ugoal)))**2 / np.abs(
np.trace(np.dot(pulseParams.Ugoal.conj().T,
pulseParams.Ugoal)))**2 > 0.99
def setup_system():
'''
Should probably take in some parameters, but for now hard-code things.
'''
#Setup the system
systemParams = SystemParams()
#First the two qubits defined in the lab frame
Q1 = SCQubit(numLevels=3, omega=4.279e9, delta=-239e6, name='Q1', T1=10e-6)
systemParams.add_sub_system(Q1)
Q2 = SCQubit(numLevels=3, omega=4.06e9, delta=-224e6, name='Q2', T1=10e-6)
systemParams.add_sub_system(Q2)
#Add an interaction between the qubits to get the the cross-resonance effect
systemParams.add_interaction('Q1', 'Q2', 'FlipFlop', 2e6)
#Create the full Hamiltonian
systemParams.create_full_Ham()
#Calculate the eigenframe for the natural Hamiltonian
d, v = eigh(systemParams.Hnat.matrix)
#Reorder the transformation matrix to maintain the computational basis ordering
sortOrder = np.argsort(np.argmax(np.abs(v), axis=0))
v = v[:, sortOrder]
systemParams.Hnat.matrix = np.complex128(np.diag(d[sortOrder]))
#Some operators for the controls
X = 0.5 * (Q1.loweringOp + Q1.raisingOp)
Y = 0.5 * (-1j * Q1.loweringOp + 1j * Q1.raisingOp)
#The cross-coupling between the drives
crossCoupling12 = 5.0 / 20
crossCoupling21 = 2.0 / 30
#Add the Q1 drive Hamiltonians
inPhaseHam = Hamiltonian(
np.dot(
v.conj().T,
np.dot(
systemParams.expand_operator('Q1', X) +
crossCoupling12 * systemParams.expand_operator('Q2', X), v)))
quadratureHam = Hamiltonian(
np.dot(
v.conj().T,
np.dot(
systemParams.expand_operator('Q1', Y) +
crossCoupling12 * systemParams.expand_operator('Q2', Y), v)))
systemParams.add_control_ham(inphase=inPhaseHam, quadrature=quadratureHam)
systemParams.add_control_ham(inphase=inPhaseHam, quadrature=quadratureHam)
#Add the cross-drive Hamiltonians (same drive line as Q1)
inPhaseHam = Hamiltonian(
np.dot(
v.conj().T,
np.dot(
systemParams.expand_operator('Q1', X) +
crossCoupling12 * systemParams.expand_operator('Q2', X), v)))
quadratureHam = Hamiltonian(
np.dot(
v.conj().T,
np.dot(
systemParams.expand_operator('Q1', Y) +
crossCoupling12 * systemParams.expand_operator('Q2', Y), v)))
systemParams.add_control_ham(inphase=inPhaseHam, quadrature=quadratureHam)
systemParams.add_control_ham(inphase=inPhaseHam, quadrature=quadratureHam)
#Add the Q2 drive Hamiltonians
inPhaseHam = Hamiltonian(
np.dot(
v.conj().T,
np.dot(
systemParams.expand_operator('Q2', X) +
crossCoupling21 * systemParams.expand_operator('Q1', X), v)))
quadratureHam = Hamiltonian(
np.dot(
v.conj().T,
np.dot(
systemParams.expand_operator('Q2', Y) +
crossCoupling21 * systemParams.expand_operator('Q1', Y), v)))
systemParams.add_control_ham(inphase=inPhaseHam, quadrature=quadratureHam)
systemParams.add_control_ham(inphase=inPhaseHam, quadrature=quadratureHam)
#Setup the measurement operator
systemParams.measurement = np.diag(
np.array([0.088, 0.082, 0.080, 0.080, 0.78, 0.75, 0.078, 0.074,
0.072]))
##Add the T1 dissipators
#systemParams.dissipators.append(Dissipator(systemParams.expand_operator('Q1', Q1.T1Dissipator)))
#systemParams.dissipators.append(Dissipator(systemParams.expand_operator('Q2', Q2.T1Dissipator)))
#
return systemParams, (Q1, Q2)
