# TopK and HierarchicalTopK SAE decoder Triton kernels
# Copyright 2025 T-Tech


from .topk_kernels import triton_topk_sae_loss, topk_sae_loss
from .hierarchical_kernels import triton_hierarchical_sae_loss, hierarchical_sae_loss

__kernel_metadata__ = {
    "license": "Apache-2.0 (with CC-BY-NC-4.0 component; see NOTICE)",
}

__all__ = [
    "__kernel_metadata__",
    "topk_sae_loss",
    "triton_topk_sae_loss",
    "hierarchical_sae_loss",
    "triton_hierarchical_sae_loss",
]