fla_layer_slidingwindowsharedkeyattention small fla layer NLP_LLM_text not forward validated unknown 15 ops
fla_lightnet_for_causal_lm small fla lightnet time-series_dynamics not forward validated unknown 12 ops
fla_linear_attention_for_causal_lm medium fla linear time-series_dynamics not forward validated unknown 50 ops
fla_log_linear_mamba2_for_causal_lm medium fla log time-series_dynamics not forward validated unknown 78 ops
fla_mamba2_for_causal_lm medium fla mamba2 time-series_dynamics not forward validated unknown 144 ops