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
221 B
Python
6 lines
221 B
Python
"""Incremental recurrent KDA implementations and state containers."""
|
|
|
|
from .fused import KDAState, fused_recurrent_kda, fused_recurrent_kda_step
|
|
|
|
__all__ = ["KDAState", "fused_recurrent_kda", "fused_recurrent_kda_step"]
|