q = project_query(x)

k = project_key(x)

q = apply_position(q)

k = apply_position(k)

attention_scores = q @ k.T