Initial K3 snapshot: 0.5B KDA/MLA/MoE train path
Standalone tree split from LLMRL/projects/kda. Includes Triton dt_bias backward fix, train_k3 --preset 0.5b, SFT, Docker runtime, and tests.
This commit is contained in:
@@ -0,0 +1,5 @@
|
||||
"""FLA ``chunk_kda`` surface used by ``ops.api`` backend='triton'."""
|
||||
|
||||
from kda._fla.ops.kda.chunk import ChunkKDAFunction, chunk_kda
|
||||
|
||||
__all__ = ["ChunkKDAFunction", "chunk_kda"]
|
||||
Reference in New Issue
Block a user