Tinygrad: Numerically Stable Cross-EntropyDeep-ML | Problem