Tinygrad: Cosine LR Schedule with Linear WarmupDeep-ML | Problem