TikTok Interview Question

Write self-attention and cross-entropy pseudocode