mirror of
https://github.com/harvard-edge/cs249r_book.git
synced 2026-04-29 17:20:21 -05:00
a78f1bd8b0
Complete MLSYSIM v0.1.0 implementation with: - Documentation website (Quarto): landing page with animated hero and capability carousel, 4 tutorials (hello world, LLM serving, distributed training, sustainability), hardware/model/fleet/infra catalogs, solver guide, whitepaper, math foundations, glossary, and full quartodoc API reference - Typed registry system: Hardware (18 devices across 5 tiers), Models (15 workloads), Systems (fleets, clusters, fabrics), Infrastructure (grid profiles, rack configs, datacenters) - Core types: Pint-backed Quantity, Metadata provenance tracking, custom exception hierarchy (OOMError, SLAViolation) - SimulationConfig with YAML/JSON loading and pre-validation - Scenario system tying workloads to systems with SLA constraints - Multi-level evaluation scorecard (feasibility, performance, macro) - Examples, tests, and Jetson Orin NX spec fix (100 → 25 TFLOP/s) Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
# mlsysim/__init__.py
|
|
"""
|
|
mlsysim: Machine Learning Systems Infrastructure and Modeling Platform
|
|
"""
|
|
|
|
from . import core
|
|
from . import hardware
|
|
from . import models
|
|
from . import infra
|
|
from . import systems
|
|
from . import sim
|
|
from . import fmt
|
|
from . import viz
|
|
|
|
# Explicitly export submodules for documentation and execution
|
|
from . import hardware as hardware_mod
|
|
from . import models as models_mod
|
|
from . import infra as infra_mod
|
|
from . import systems as systems_mod
|
|
|
|
# Export primary API objects for convenience
|
|
from .hardware.types import HardwareNode
|
|
from .models.types import Workload, TransformerWorkload, CNNWorkload
|
|
from .systems.types import Fleet, Node, NetworkFabric, DeploymentTier
|
|
from .core.scenarios import Scenario, Scenarios, Applications
|
|
from .core.engine import Engine
|
|
from .core.config import SimulationConfig, load_config
|
|
from .core.solver import (
|
|
SingleNodeSolver,
|
|
DistributedSolver,
|
|
ReliabilitySolver,
|
|
SustainabilitySolver,
|
|
EconomicsSolver,
|
|
ServingSolver
|
|
)
|
|
|
|
# Export Registries
|
|
from .hardware.registry import Hardware
|
|
from .models.registry import Models
|
|
from .infra.registry import Infra
|
|
from .systems.registry import Systems
|
|
|
|
# Export unit registry for custom workload definitions
|
|
from .core.constants import ureg
|
|
|
|
# Visualization
|
|
from .viz.plots import plot_evaluation_scorecard, plot_roofline
|