dela
584f7e9e73
Standalone tree split from LLMRL/projects/kda. Includes Triton dt_bias backward fix, train_k3 --preset 0.5b, SFT, Docker runtime, and tests.
6 lines
179 B
Python
6 lines
179 B
Python
"""FLA ``chunk_kda`` surface used by ``ops.api`` backend='triton'."""
|
|
|
|
from kda._fla.ops.kda.chunk import ChunkKDAFunction, chunk_kda
|
|
|
|
__all__ = ["ChunkKDAFunction", "chunk_kda"]
|