Pre-Norm GPT Transformer Block Forward PassDeep-ML | Problem