Concept: Instead of importing a heavy transformer model, have the user implement the core mathematical engine of the transformer block using basic matrix manipulation.
Implement a single-head Scaled Dot-Product Attention mechanism. Given matrices Q (Queries), K (Keys), and V(Values), calculate the attention outputs.
Constraints Allowed libraries: numpy
The softmax function must be numerically stable (handling potential overflow).