1#![forbid(unsafe_code)]
8#![warn(missing_docs)]
9
10pub mod artifact;
12pub mod attention;
14pub mod automatic;
16pub mod backend;
18pub mod cache;
20pub mod capability;
22pub mod checkpoint;
24pub mod consensus;
26pub mod execution;
28pub mod generation;
30pub mod input;
32pub mod inspection;
34pub mod media;
36pub mod observation;
38pub mod realtime;
40pub mod residency;
42pub mod scheduler;
44pub mod speculative;
46pub mod topology;
48
49pub use artifact::{
50 gguf_u32_metadata_values, inspect_artifact, plan_model_preparation, resolve_gguf_companions,
51 validate_preparation_policy, ArtifactFormat, ArtifactInspection, GgufCompanionEncoding,
52 GgufCompanionRequirement, GgufCompanionRole, LoadingProtocol, MaterializationRoute,
53 ModelArtifact, ModelConfiguration, ModelConfigurationResolver, ModelPreparationPlan,
54 PreparationPolicy, QuantizationRequest, ResidencyRequest, ResolvedModelConfiguration,
55 ValidatedGguf, ValidatedGgufCompanion,
56};
57pub use attention::{AttentionPolicy, LayerSchedule, LayerScheduleError};
58pub use automatic::{
59 realize_execution_plan_drafting, realize_execution_plan_target, AllocatorTelemetry,
60 AutomaticPlanRequest, AutomaticPlanner, AutomaticPlannerPolicy, AutomaticPlanningBackend,
61 AutomaticPlanningError, BoundedResidencyRequirement, CandidateAdmission, DurationSeconds,
62 ExecutionPlanBackendFactory, ExecutionPlanReport, ExecutionPlanTarget, ExecutionTelemetry,
63 ExpertCacheTelemetry, ExternalDraftArtifact, HardwareBackendProfile, HardwareDeviceProfile,
64 HardwareMemorySemantics, HardwareProfile, ModelResourceProfile, ObservationKind, Observed,
65 PlanExplanation, PlanExplanationEntry, PlanExplanationLevel, RealizedDrafting,
66 ResidencyTelemetry, SpeculativeDecodingTelemetry, TimingTelemetry, TokenizerCompatibilityError,
67 TokenizerCompatibilityProof, TransferTelemetry, AUTOMATIC_SCHEMA_VERSION,
68};
69pub use backend::{
70 load_model, prepare_inspected_model, BackendDescriptor, BackendError, BackendProvider,
71 BackendSession, CollectiveGroupDescriptor, CollectiveGroupId, CollectiveScope, Completion,
72 ControlledTextGeneration, ControlledTextGenerationError, ControlledToken, DeviceCapabilities,
73 DeviceDescriptor, DistributedBackend, DistributedCapabilities, DistributedSession,
74 DistributedSessionDescriptor, InspectableBackendSession, ModelCapabilityBackend,
75 ModelLoadError, ModelLoadingBackend, ModelRuntime, MultimodalPreparationBackend,
76 MultimodalPreparationFailure, PreparedModel, SessionCapabilities, SessionCapabilityError,
77 SpeculativeTokenFilterController, Submission, TextGeneration, TextGenerationBackend,
78 TextGenerationConfig, TextSamplingStrategy, TokenFilter, TokenFilterController,
79 TokenFilterError, TokenOutput, ValueDescriptor,
80};
81pub use capability::{
82 apply_admission_policy, estimate_runtime_state, Admission, AdmissionRejection,
83 AdmissionRequest, AdmissionResult, AvailableMemory, CacheStateStrategy, CapabilityError,
84 EstimationCompleteness, InputModalities, InputTokenCount, ModelCapabilities,
85 PhysicalMemorySemantics, RuntimeStateEstimate, SlidingWindowLayerCount, StateMemoryAssumptions,
86 StateMemoryLayout, StaticMemoryReport,
87};
88pub use execution::{
89 BackendId, DevicePlan, DraftPlacementPlan, DraftingPlan, ExecutionPlan, ExecutionPlanError,
90 ExpertCachePlan, ResidencyPlan, WeightTransformationPlan, DEFAULT_MAX_CACHED_SHARDS,
91 EXECUTION_PLAN_SCHEMA_VERSION,
92};
93pub use generation::{
94 resolve_generation_config, resolve_optimistic_reuse, CheckpointGenerationConfig, FinishReason,
95 GenerationCancellationToken, GenerationConfigOverrides, GenerationError, GenerationSequence,
96 OptimisticReuseDecision, ResolvedGenerationConfig, SemanticEvent,
97 SpeculativeCancellationDisposition, SpeculativeCommitPlan, SpeculativeConfig,
98 SpeculativeRequestId, SpeculativeRequestLifecycle, SpeculativeRequestStatus, SpeculativeRound,
99 SpeculativeSchedulerOptions, SpeculativeTail, TokenCommit, TokenTerminalSignals,
100};
101pub use input::{
102 InputExtent, InputMetadataKey, InputModality, InputPartDescriptor, InputPayloadKind,
103 InputTensorIdentity, PreparedInputError, PreparedInputIdentity,
104};
105pub use inspection::{
106 ArtifactModality, ArtifactTensorEncoding, InspectionIssue, InspectionIssueCode,
107 InspectionReadiness, InspectionRequirement, InspectionSeverity, ModelInspectionReport,
108};
109pub use media::{
110 Audio, Media, MediaBinding, MediaRequestError, MultimodalRequest, MultimodalSegment, RgbImage,
111 TokenizedMultimodalRequest, TokenizedMultimodalSegment, Video, VideoSampling,
112};
113pub use observation::{
114 InspectedOutput, ObservationError, ObservationRequest, ObservationSelector, ObservationSet,
115 ObservationValue, TensorObservation, TensorObservationData, MODEL_LOGITS_OBSERVATION_PATH,
116};
117pub use realtime::{
118 load_realtime_model, load_realtime_model_with_options, RealtimeBackend, RealtimeCompletedStep,
119 RealtimeConfigError, RealtimeDecisionDiagnostics, RealtimeError, RealtimeFrameConvention,
120 RealtimeFrameForcing, RealtimeFrameScheduleState, RealtimeFrameSlot, RealtimeFrameTransition,
121 RealtimeInputFrame, RealtimeModel, RealtimeModelLoadingBackend, RealtimeOutputFrame,
122 RealtimeSampling, RealtimeScheduleError, RealtimeScheduler, RealtimeSession,
123 RealtimeSlotCoordinate, RealtimeSlotOccupancy, RealtimeSpeechConfig, RealtimeTargetDecision,
124 RealtimeTargetSource, RealtimeTemporalSource, MAX_REALTIME_FRAME_DELAY,
125};
126pub use residency::{
127 BackgroundPrefetchReport, PrefetchAdmission, PrefetchCompletion, PrefetchDemandObservation,
128 PrefetchDemandResolution, PrefetchExecutionState, PrefetchStateError, PrefetchWork,
129};
130pub use speculative::{
131 cancel_pending_verification, propose_block, resolve_commit_and_publish,
132 resolve_optimistic_branch, resolve_round, submit_verification_transaction,
133 CompletedSpeculativeRequest, CompletedSpeculativeSchedule, PendingSpeculativeVerification,
134 PreparedSpeculativeLane, ProposalDecision, PublishedSpeculativeResult,
135 PublishedSpeculativeVerification, ResolvedSpeculativeRound, SamplingPlacement,
136 SpeculativeAction, SpeculativeCallbackPublisher, SpeculativeCandidate, SpeculativeCapability,
137 SpeculativeCommit, SpeculativeConstraint, SpeculativeContinuation, SpeculativeDraft,
138 SpeculativeDraftBlock, SpeculativeDraftSource, SpeculativeDriverError,
139 SpeculativeExecutionTopology, SpeculativeExecutor, SpeculativeGenerationBackend,
140 SpeculativeGenerationBatchOutput, SpeculativeGenerationBatchRequest, SpeculativeGenerationLane,
141 SpeculativeGenerationOutput, SpeculativeGenerationVisitor, SpeculativeOptimisticBranch,
142 SpeculativeOutputError, SpeculativeOutputRuntime, SpeculativePrefill, SpeculativeProposal,
143 SpeculativePublicationStatus, SpeculativePublisher, SpeculativeRandomness, SpeculativeRequest,
144 SpeculativeRequestTable, SpeculativeSampling, SpeculativeSchedule, SpeculativeSchedulerStats,
145 SpeculativeSemanticConstraint, SpeculativeSemanticState, SpeculativeStats,
146 SpeculativeTelemetry,
147};
148pub use topology::{
149 balanced_contiguous_range, ParallelAxis, ParallelCoordinates, ParallelRankTopology,
150 ParallelTopology, SubgroupMembership, TopologyError, TopologyPreflightReport,
151};