TikTok Interview Question

Write cross entropy loss and adamW optimizer