pub fn softcapped_window_attention_f32(
query: &[f32],
key: &[f32],
value: &[f32],
batch: usize,
query_len: usize,
key_len: usize,
heads: usize,
kv_heads: usize,
head_dim: usize,
scale: f32,
causal: bool,
window_size: usize,
soft_cap: Option<f32>,
) -> Vec<f32>