File size: 483 Bytes
a262a48 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
# 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",
]
|