This section is at https://rollingedition.com/language-models-from-the-ground-up/training-a-small-model#update-with-adamw.