Tinygrad: Implement a Transformer Encoder BlockDeep-ML | Problem