Read and refactor a large PyTorch training file
You get a few hundred lines of existing PyTorch code that trains a classifier. Answer questions about its time and memory cost, then make a series of changes to it in parts, keeping it working after each one. Reading unfamiliar code quickly matters more than writing new code.
No answer yet. Write the first one; the best answer gets endorsed.