Trajectree 0.0.1__py3-none-any.whl → 0.0.3__py3-none-any.whl

trajectree/quimb/docs/examples/ex_mpi_expm_evo.py

@@ -1,62 +0,0 @@
-"""This script illustrates how quimb's various MPI modes work.
-
-It can be run as:
-
-python ex_mpi_modes_expm_evo.py
-quimb-mpi-python ex_mpi_modes_expm_evo.py
-quimb-mpi-python --syncro ex_mpi_modes_expm_evo.py
-
-And will display slightly different output for each that explains the
-three modes.
-"""
-import quimb as qu
-from mpi4py import MPI
-
-# Get some MPI information
-comm = MPI.COMM_WORLD
-rank, size = comm.Get_rank(), comm.Get_size()
-print(f"I am worker {rank} of total {size} running main script...")
-
-
-# setup a verbose version of the ham_heis constructor, and make it Lazy
-n = 18
-shape = (2**n, 2**n)
-
-# this makes the function print some information when called
-#     - in order to be pickled is has to be located in the main package
-ham_heis_verbose = qu.utils.Verbosify(qu.ham_heis,
-                                      highlight='ownership', mpi=True)
-
-H = qu.Lazy(ham_heis_verbose, n=n, sparse=True, shape=shape)
-
-# random initial state
-#     - must make sure all processes have the same seed to be pure
-psi0 = qu.rand_ket(2**n, seed=42)
-
-# evolve the system, processes split 'hard' work (slepc computations)
-#     - should see each worker gets given a different ownership rows
-#     - but all end up with the results.
-evo = qu.Evolution(psi0, H, method='expm', expm_backend='slepc')
-evo.update_to(5)
-
-
-print(f"{rank}: I have final state norm {qu.expec(evo.pt, evo.pt)}")
-
-# Now lets demonstrate using the MPI pool construct
-pool = qu.get_mpi_pool()
-
-dims = [2] * n
-bsz = 5
-logneg_subsys_verbose = qu.utils.Verbosify(qu.logneg_subsys,
-                                           highlight='sysb', mpi=True)
-
-# each process only computes its own fraction of these
-#     - should see each process calls logneg with different ``sysb``.
-fs = [pool.submit(logneg_subsys_verbose, evo.pt, dims=dims,
-                  sysa=range(0, bsz), sysb=range(i, i + bsz))
-      for i in range(bsz, n - bsz)]
-
-# but then the results are comminucated to everyone
-rs = [f.result() for f in fs]
-
-print(f"{rank}: I have logneg results: {rs}")

trajectree/quimb/quimb/__init__.py

@@ -1,507 +0,0 @@
-"""
-Quantum Information for Many-Body calculations.
-"""
-try:
-    # -- Distribution mode --
-    # import from _version.py generated by setuptools_scm during release
-    from ._version import version as __version__
-except ImportError:
-    # -- Source mode --
-    try:
-        # use setuptools_scm to get the current version from src using git
-        from setuptools_scm import get_version as _gv
-        from pathlib import Path as _Path
-
-        __version__ = _gv(_Path(__file__).parent.parent)
-    except ImportError:
-        # setuptools_scm is not available, use a default version
-        __version__ = "0.0.0+unknown"
-
-import warnings
-
-# some useful math
-from math import pi, cos, sin, tan, exp, log, log2, log10, sqrt
-
-# Core functions
-from .core import (
-    qarray,
-    prod,
-    isket,
-    isbra,
-    isop,
-    isvec,
-    issparse,
-    isdense,
-    isreal,
-    isherm,
-    ispos,
-    mul,
-    dag,
-    dot,
-    vdot,
-    rdot,
-    ldmul,
-    rdmul,
-    outer,
-    explt,
-    get_thread_pool,
-    normalize,
-    chop,
-    quimbify,
-    qu,
-    ket,
-    bra,
-    dop,
-    sparse,
-    infer_size,
-    trace,
-    identity,
-    eye,
-    speye,
-    dim_map,
-    dim_compress,
-    kron,
-    kronpow,
-    ikron,
-    pkron,
-    permute,
-    itrace,
-    partial_trace,
-    expectation,
-    expec,
-    nmlz,
-    tr,
-    ptr,
-)
-
-# Linear algebra functions
-from .linalg.base_linalg import (
-    eigensystem,
-    eig,
-    eigh,
-    eigvals,
-    eigvalsh,
-    eigvecs,
-    eigvecsh,
-    eigensystem_partial,
-    groundstate,
-    groundenergy,
-    bound_spectrum,
-    eigh_window,
-    eigvalsh_window,
-    eigvecsh_window,
-    svd,
-    svds,
-    norm,
-    expm,
-    sqrtm,
-    expm_multiply,
-    Lazy,
-)
-from .linalg.rand_linalg import rsvd, estimate_rank
-from .linalg.mpi_launcher import get_mpi_pool, can_use_mpi_pool
-
-# Generating objects
-from .gen.operators import (
-    spin_operator,
-    pauli,
-    hadamard,
-    phase_gate,
-    S_gate,
-    T_gate,
-    U_gate,
-    rotation,
-    Rx,
-    Ry,
-    Rz,
-    Xsqrt,
-    Ysqrt,
-    Zsqrt,
-    Wsqrt,
-    swap,
-    iswap,
-    fsim,
-    fsimg,
-    ncontrolled_gate,
-    controlled,
-    CNOT,
-    cX,
-    cY,
-    cZ,
-    ccX,
-    ccY,
-    ccZ,
-    controlled_swap,
-    cswap,
-    fredkin,
-    toffoli,
-    ham_heis,
-    ham_ising,
-    ham_XY,
-    ham_XXZ,
-    ham_j1j2,
-    ham_mbl,
-    ham_heis_2D,
-    zspin_projector,
-    create,
-    destroy,
-    num,
-    ham_hubbard_hardcore,
-)
-from .gen.states import (
-    basis_vec,
-    up,
-    zplus,
-    down,
-    zminus,
-    plus,
-    xplus,
-    minus,
-    xminus,
-    yplus,
-    yminus,
-    bloch_state,
-    bell_state,
-    singlet,
-    thermal_state,
-    neel_state,
-    singlet_pairs,
-    werner_state,
-    ghz_state,
-    w_state,
-    levi_civita,
-    perm_state,
-    graph_state_1d,
-    computational_state,
-)
-from .gen.rand import (
-    randn,
-    rand,
-    rand_matrix,
-    rand_herm,
-    rand_pos,
-    rand_rho,
-    rand_ket,
-    rand_uni,
-    rand_haar_state,
-    gen_rand_haar_states,
-    rand_mix,
-    rand_product_state,
-    rand_matrix_product_state,
-    rand_mps,
-    rand_seperable,
-    rand_iso,
-    rand_mera,
-    seed_rand,
-    set_rand_bitgen,
-)
-
-# Functions for calculating properties
-from .calc import (
-    fidelity,
-    purify,
-    entropy,
-    entropy_subsys,
-    mutual_information,
-    mutinf,
-    mutinf_subsys,
-    schmidt_gap,
-    tr_sqrt,
-    tr_sqrt_subsys,
-    partial_transpose,
-    negativity,
-    logarithmic_negativity,
-    logneg,
-    logneg_subsys,
-    concurrence,
-    one_way_classical_information,
-    quantum_discord,
-    trace_distance,
-    cprint,
-    decomp,
-    pauli_decomp,
-    bell_decomp,
-    correlation,
-    pauli_correlations,
-    ent_cross_matrix,
-    qid,
-    is_degenerate,
-    is_eigenvector,
-    page_entropy,
-    heisenberg_energy,
-    dephase,
-    kraus_op,
-    projector,
-    measure,
-    simulate_counts,
-)
-
-# Evolution class and methods
-from .evo import Evolution
-
-from .linalg.approx_spectral import (
-    approx_spectral_function,
-    tr_abs_approx,
-    tr_exp_approx,
-    tr_sqrt_approx,
-    tr_xlogx_approx,
-    entropy_subsys_approx,
-    logneg_subsys_approx,
-    negativity_subsys_approx,
-    xlogx,
-)
-from .utils import (
-    save_to_disk,
-    load_from_disk,
-    oset,
-    LRU,
-    tree_map,
-    tree_apply,
-    tree_flatten,
-    tree_unflatten,
-    format_number_with_error,
-    NEUTRAL_STYLE,
-    default_to_neutral_style,
-)
-
-
-warnings.filterwarnings("ignore", message="Caching is not available when ")
-
-
-__all__ = [
-    # Accel ----------------------------------------------------------------- #
-    "qarray",
-    "prod",
-    "isket",
-    "isbra",
-    "isop",
-    "isvec",
-    "issparse",
-    "isdense",
-    "isreal",
-    "isherm",
-    "ispos",
-    "mul",
-    "dag",
-    "dot",
-    "vdot",
-    "rdot",
-    "ldmul",
-    "rdmul",
-    "outer",
-    "explt",
-    # Core ------------------------------------------------------------------ #
-    "normalize",
-    "chop",
-    "quimbify",
-    "qu",
-    "ket",
-    "bra",
-    "dop",
-    "sparse",
-    "infer_size",
-    "trace",
-    "identity",
-    "eye",
-    "speye",
-    "dim_map",
-    "dim_compress",
-    "kron",
-    "kronpow",
-    "ikron",
-    "pkron",
-    "permute",
-    "itrace",
-    "partial_trace",
-    "expectation",
-    "expec",
-    "nmlz",
-    "tr",
-    "ptr",
-    # Linalg ---------------------------------------------------------------- #
-    "eigensystem",
-    "eig",
-    "eigh",
-    "eigvals",
-    "eigvalsh",
-    "eigvecs",
-    "eigvecsh",
-    "eigensystem_partial",
-    "groundstate",
-    "groundenergy",
-    "bound_spectrum",
-    "eigh_window",
-    "eigvalsh_window",
-    "eigvecsh_window",
-    "svd",
-    "svds",
-    "norm",
-    "Lazy",
-    "rsvd",
-    "estimate_rank",
-    # Gen ------------------------------------------------------------------- #
-    "spin_operator",
-    "pauli",
-    "hadamard",
-    "phase_gate",
-    "T_gate",
-    "S_gate",
-    "U_gate",
-    "rotation",
-    "Rx",
-    "Ry",
-    "Rz",
-    "Xsqrt",
-    "Ysqrt",
-    "Zsqrt",
-    "Wsqrt",
-    "swap",
-    "iswap",
-    "fsim",
-    "fsimg",
-    "ncontrolled_gate",
-    "controlled",
-    "CNOT",
-    "cX",
-    "cY",
-    "cZ",
-    "ccX",
-    "ccY",
-    "ccZ",
-    "controlled_swap",
-    "cswap",
-    "fredkin",
-    "toffoli",
-    "ham_heis",
-    "ham_ising",
-    "ham_XY",
-    "ham_XXZ",
-    "ham_j1j2",
-    "ham_mbl",
-    "ham_heis_2D",
-    "create",
-    "destroy",
-    "num",
-    "ham_hubbard_hardcore",
-    "zspin_projector",
-    "basis_vec",
-    "up",
-    "zplus",
-    "down",
-    "zminus",
-    "plus",
-    "xplus",
-    "minus",
-    "xminus",
-    "yplus",
-    "yminus",
-    "bloch_state",
-    "bell_state",
-    "singlet",
-    "thermal_state",
-    "neel_state",
-    "singlet_pairs",
-    "werner_state",
-    "ghz_state",
-    "w_state",
-    "levi_civita",
-    "perm_state",
-    "graph_state_1d",
-    "rand_matrix",
-    "rand_herm",
-    "rand_pos",
-    "rand_rho",
-    "rand_ket",
-    "rand_uni",
-    "rand_haar_state",
-    "gen_rand_haar_states",
-    "rand_mix",
-    "rand_mps",
-    "randn",
-    "rand",
-    "rand_product_state",
-    "rand_matrix_product_state",
-    "rand_seperable",
-    "rand_iso",
-    "rand_mera",
-    "seed_rand",
-    "set_rand_bitgen",
-    "computational_state",
-    # Calc ------------------------------------------------------------------ #
-    "expm",
-    "sqrtm",
-    "expm_multiply",
-    "fidelity",
-    "purify",
-    "entropy",
-    "entropy_subsys",
-    "mutual_information",
-    "mutinf",
-    "mutinf_subsys",
-    "schmidt_gap",
-    "tr_sqrt",
-    "tr_sqrt_subsys",
-    "partial_transpose",
-    "negativity",
-    "logarithmic_negativity",
-    "logneg",
-    "logneg_subsys",
-    "concurrence",
-    "one_way_classical_information",
-    "quantum_discord",
-    "trace_distance",
-    "cprint",
-    "decomp",
-    "pauli_decomp",
-    "bell_decomp",
-    "correlation",
-    "pauli_correlations",
-    "ent_cross_matrix",
-    "qid",
-    "is_degenerate",
-    "is_eigenvector",
-    "page_entropy",
-    "heisenberg_energy",
-    "dephase",
-    "kraus_op",
-    "projector",
-    "measure",
-    "simulate_counts",
-    # Evo ------------------------------------------------------------------- #
-    "Evolution",
-    # Approx spectral ------------------------------------------------------- #
-    "approx_spectral_function",
-    "tr_abs_approx",
-    "tr_exp_approx",
-    "tr_sqrt_approx",
-    "tr_xlogx_approx",
-    "entropy_subsys_approx",
-    "logneg_subsys_approx",
-    "negativity_subsys_approx",
-    # Some misc useful math ------------------------------------------------- #
-    "pi",
-    "cos",
-    "sin",
-    "tan",
-    "exp",
-    "log",
-    "log2",
-    "log10",
-    "sqrt",
-    "xlogx",
-    # Utils ----------------------------------------------------------------- #
-    "save_to_disk",
-    "load_from_disk",
-    "get_thread_pool",
-    "get_mpi_pool",
-    "can_use_mpi_pool",
-    "oset",
-    "LRU",
-    "tree_map",
-    "tree_apply",
-    "tree_flatten",
-    "tree_unflatten",
-    "format_number_with_error",
-    "NEUTRAL_STYLE",
-    "default_to_neutral_style",
-]