Unrolling thread...
Causal Masking in Attention