tsu-emulator/tsu.py at main · Arsham-001/tsu-emulator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
"""
TSU (Thermodynamic Sampling Unit) - Main Package
Re-exports public API from tsu_api for easy importing

Usage:
    import tsu
    samples = tsu.sample_gaussian(mu=0, sigma=1, n=1000)
"""

# Import all public classes and functions from tsu_api
from tsu_api import (
    # Enums
    Backend,
    # Data classes
    SamplingResult,
    # Base classes
    Sampler,
    OptimizationProblem,
    ProbabilisticLayer,
    ProbabilisticModel,
    # Samplers
    GaussianSampler,
    MultimodalSampler,
    BayesianSampler,
    # Optimization
    MaxCutProblem,
    # Layers
    StochasticLinear,
    BernoulliActivation,
    # Functions
    sample_gaussian,
    sample_multimodal,
    compare_samplers,
    optimize,
)

# Version info
__version__ = "0.1.0"
__author__ = "TSU Research Team"

# Public API
__all__ = [
    # Enums
    "Backend",
    # Data classes
    "SamplingResult",
    # Base classes
    "Sampler",
    "OptimizationProblem",
    "ProbabilisticLayer",
    "ProbabilisticModel",
    # Samplers
    "GaussianSampler",
    "MultimodalSampler",
    "BayesianSampler",
    # Optimization
    "MaxCutProblem",
    # Layers
    "StochasticLinear",
    "BernoulliActivation",
    # Functions
    "sample_gaussian",
    "sample_multimodal",
    "compare_samplers",
    "optimize",
]