Looped Transformer Stack Forward PassDeep-ML | Problem