3. Calculating Quantities

There are various built-in functions to calculate quantities, not limited to:

3.1. Approximate Spectral Functions

The module approx_spectral, contains a Lanczos method for estimating any quantities of the form tr(fn(A)). Where A is any operator that implements a dot product with a vector. For example, estimating the trace of the sqrt of a matrix would naievly require diagonalising it:

import quimb as qu

rho = qu.rand_rho(2**12)
rho_el = qu.eigvalsh(rho)
sum(rho_el ** 0.5)
54.321684719154966
qu.tr_sqrt_approx(rho)
54.00283979240904

Diagonalization has a cost of O(n^3), which is essentially reduced to O(k * n^2) for this stochastic method. For a general function approx_spectral_function() can be used.

However, the real advantage occurs when the full matrix does not need to be fully represented, e.g. in the case of ‘partial trace states’. One can then calculate quantities for subsystems that would not be possible to explicitly represent.

For example, the partial trace, followed by partial transpose, followed by vector multiplication can be ‘lazily’ evaluated as a tensor contraction (see lazy_ptr_ppt_dot()). In this way the logarithmic negativity of subsytems can be efficiently calculated:

psi = qu.rand_ket(2**20)
dims = [2**8, 2**4, 2**8]
qu.logneg_subsys_approx(psi, dims, sysa=0, sysb=2)
5.743478523097124

Under the hood, this method makes use of quimb.tensor functionality, which allows various tensor contraction backends to be used (see set_tensor_linop_backend()). These types of computation are particularly suited to the GPU and therefore if cupy is installed it will be used automatically:

%timeit qu.logneg_subsys_approx(psi, dims, sysa=0, sysb=2, backend='numpy')
2.21 s ± 194 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)
%timeit qu.logneg_subsys_approx(psi, dims, sysa=0, sysb=2, backend='cupy')
111 ms ± 6.78 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)