Sample code for Swift Talk 495: Building a Language Model – Gradient Descent - View it on GitHub