LLM Systems Assignment 3: Transformer Architecture
本次 homework 的任务是实现 Decoder-only 的 Transformer 模型,也就是
GPT-2,并在一个翻译任务数据集上进行模型训练。具体来说,包括以下几个子任务:
实现 CrossEntropy Loss
实现基础模块,包括 Linear/Dropout/LayerNorm1d/Embedding
实现
Transformer,包括:MultiHeadAtte