
Machine Learning Engineer Interview Experience
Interview process
The system design interview was data collection, SFT and RL stages. It went well for me. It was focused on LLM post training techniques. The ML coding was worst for me although I practiced the same question but using PyTorch and doing without PyTorch and without any documentation was challenging for me.
- Recruiter screen
- Technical interview
- Take-home project
Interview tips
Brush up numpy knowledge. Every round would be on LLM or related to it. They focus heavily on LLM. The paper presentation is unique in the process for this type of role. Talk to different PhD candidates on how to present a paper well.
Company culture
I liked that the team was focused on post training. You need to work with the frontier model team which is cool and has lot of scope for learning. The remote culture attracted me.
Questions asked
Question types asked
Specific questions asked
Decode tokens from LLM using top K sampling without using PyTorch
Design knowledge distillation of models. Design LLM post training pipeline.
Get full access with a membership, or share your experience to try it free.