TikTok Interview Question

implement a transformer using Pytorch/tensorflow