| # 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", | |
| ] | |