Machine Learning Engineer Intern Interview Questions

230 machine learning engineer intern interview questions shared by candidates

Implement Single head and Multi head attention in pytorch. A follow up to this was use masking to mask out the later-on tokens. Also asked me to implement KL divergence in python (but maybe this was because I talked about kl-divergence in my projects)
avatar

Machine Learning Engineer Intern

Interviewed at TikTok

3.2
Dec 4, 2025

Implement Single head and Multi head attention in pytorch. A follow up to this was use masking to mask out the later-on tokens. Also asked me to implement KL divergence in python (but maybe this was because I talked about kl-divergence in my projects)

Viewing 131 - 140 interview questions

Glassdoor has 230 interview questions and reports from Machine learning engineer intern interviews. Prepare for your interview. Get hired. Love your job.