quemb.molbe.sparse_2el_integralΒΆ

Functions

account_for_symmetry(reachable)

Account for permutational symmetry and remove all q that are larger than p.

conversions_AO_shell(mol)

Return dictionaries that for a shell index return the corresponding AO indices and for an AO index return the corresponding shell index.

find_screening_radius(mol[, auxmol, ...])

Return a dictionary with radii for each element in mol that can be used to screen the 2-electron integrals to be lower than threshold.

get_atom_per_AO(mol)

get_blocks(reachable)

Return the value of the border elements of contiguous blocks in the sequence X."

get_complement(reachable)

Return the orbitals that cannot be reached by an orbital after screening.

get_dense_integrals(ints_3c2e, df_coef)

Compute dense ERIs from sparse 3-center integrals and sparse DF coefficients.

get_orbs_per_atom(atom_per_orb)

get_orbs_reachable_by_atom(orb_per_atom, ...)

get_orbs_reachable_by_orb(atom_per_orb, ...)

get_reachable(mol, atoms_per_orb[, ...])

Return the sorted orbitals that can by reached for each orbital after screening.

get_sparse_DF_integrals(mol, auxmol[, ...])

Return the 3-center 2-electron integrals in a sparse format with density fitting

identify_contiguous_blocks(X)

Identify the indices of contiguous blocks in the sequence X.

to_numba_input(exch_reachable)

Convert the reachable orbitals to a list of numpy arrays.

traverse_nonzero(g[, unique])

Traverse the non-zero elements of a semi-sparse 3-index tensor.

Classes

MutableSemiSparse3DTensor(shape, exch_reachable)

Special datastructure for semi-sparse and partially symmetric 3-indexed tensors.

SemiSparse3DTensor(unique_dense_data, shape, ...)

Special datastructure for semi-sparse and partially symmetric 3-indexed tensors.

SemiSparseSym3DTensor(unique_dense_data, ...)

Special datastructure for semi-sparse and partially symmetric 3-indexed tensors.

SparseInt2()

Sparsely stores the 2-electron integrals using chemist's notation.