File size: 1,002 Bytes
5ccb4fd | 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 | # Copyright (c) 2026 Simulacra Research Inc.
# SPDX-License-Identifier: Apache-2.0
from .api import (
batch_context,
route_context,
route_state,
select_frozen_route,
strip_router,
)
from .baseline import snis_mode_baseline
from .compiled import bind_router_kernel, compile_router_static
from .decode import (
GLOBAL_BEAM_WIDTH,
ROUTE_SAMPLES,
build_beam16,
build_route_sampler,
)
from .permutation import permute_ctx_prefix, permute_multi_ctx_prefix, permute_q_prefix
from .state import rebase_cold_samples, reframe_state_context
__all__ = [
"batch_context",
"GLOBAL_BEAM_WIDTH",
"ROUTE_SAMPLES",
"bind_router_kernel",
"build_beam16",
"build_route_sampler",
"compile_router_static",
"permute_ctx_prefix",
"permute_multi_ctx_prefix",
"permute_q_prefix",
"rebase_cold_samples",
"reframe_state_context",
"route_context",
"route_state",
"select_frozen_route",
"snis_mode_baseline",
"strip_router",
]
|