File size: 614 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 | # Copyright (c) 2026 Simulacra Research Inc.
# SPDX-License-Identifier: Apache-2.0
from .api import (
KFACBundle,
apply_finetune_kfac_step,
apply_router_kfac_step,
init_finetune_kfac_state,
init_router_kfac_state,
learning_rate,
process_finetune_targets,
process_route_targets,
register_scale_and_shift,
)
__all__ = [
"KFACBundle",
"apply_finetune_kfac_step",
"apply_router_kfac_step",
"init_finetune_kfac_state",
"init_router_kfac_state",
"learning_rate",
"process_finetune_targets",
"process_route_targets",
"register_scale_and_shift",
]
|