"""Readable PyTorch implementations used as correctness references."""

from .chunkwise import naive_chunk_kda
from .gate import kda_gate_naive, kda_gate_reference
from .recurrent import naive_kda, naive_kda_fwd

__all__ = [
    "kda_gate_naive",
    "kda_gate_reference",
    "naive_chunk_kda",
    "naive_kda",
    "naive_kda_fwd",
]