This folder contains annotated reading material for the lab page.
Separate which training states are sharded: optimizer, gradients, parameters.
Running code is optional. The expected outcome is that you can explain the mechanism, the relevant state/shape, the common misunderstanding, and the interview answer.