supergrad.time_evolution.sesolve_final_states_w_basis_trans

supergrad.time_evolution.sesolve_final_states_w_basis_trans#

supergrad.time_evolution.sesolve_final_states_w_basis_trans(hamiltonian, psi_list, tlist, transform_matrix=None, **kwargs)[source]#

Compute the final states and applying basis transformation according to the transform_matrix. The time evolution is computed using sesolve in the product basis.

Parameters:

  • hamiltonian – jax.numpy.ndarray, list System Hamiltonian as a ndarray, list of ndarray and coefficient. List format is the same as qutip.sesolve.

  • psi_list – jax.numpy.ndarray Initial state vector (ket) or array of state vectors.

  • tlist – array_like of float List of times for \(t\).

  • transform_matrix – jax.numpy.ndarray The transform unitary from product basis to the desired basis.

  • kwargs – Keyword arguments will be pass to supergrad.time_evolution.sesolve

Contents