Sample code for Swift Talk 497: Building a Language Model – MLX - View it on GitHub