feat(kda): integrate KDA attention with tokamax backend and CP support - #1

Open

chiaotung97 wants to merge 9 commits into

mainfrom

feature_kda_integration

Commits on Aug 31, 2026

feat(kda): integrate KDA attention with tokamax backend and CP support
chiaotung97
committed
fba3fbd
fix: address PR review - terminology, guard removal, doc updates
chiaotung97
committed
b039860
fix: address remaining review findings
chiaotung97
committed
b86419f
fix: pyink formatting and doc consistency
chiaotung97
committed
a1297bf
fix(kda): use mosaic implementation name, mark UTs tpu_only, add e2e smoke
chiaotung97
committed
ecc2171
fix(kda): align layer and kernel API with final tokamax kimi_delta_attention
chiaotung97
committed
a55d2f4
fix(kda): address review feedback
chiaotung97
committed
82a9fc1

Commits on Sep 2, 2026