pub const KDA_L2_EPS: f32 = 1e-6;
FLA use_qk_l2norm_in_kernel eps. CUDA k3_kda_recurrent_step_f32 uses the same.
use_qk_l2norm_in_kernel
k3_kda_recurrent_step_f32