bmak2-nlp / src / experiments / causal-language-modeling / wikipedia / depthwise-attention