Machine Learning Engineer · Phone Screen · Coding

Amazon — Machine Learning Engineer ✅ Passed

Level: Mid-Level

Round: Phone Screen · Type: Coding · Difficulty: 6/10 · Duration: 60 min · Interviewer: Neutral

Topics: Large Language Models (LLM), LoRA, Reinforcement Learning, KV Cache, Positional Encoding, Flash Attention

Location: San Francisco Bay Area

Interview date: 2025-12-31

Summary

Round 1: Phone Screen

Question: The interviewer asked about my resume projects, specifically focusing on LLM-related content. The interviewer then proceeded to ask a coding question.

Details

I was asked about two projects on my resume, including details about the data used, potential problems during model training, and how to design the application if the use case were extended.

I was also asked about:

• LoRA and its variations
• Reinforcement Learning (RL) concepts
• KV cache
• Positional encoding
• Flash attention

The coding question I received was LeetCode 135. I provided the optimal solution and discussed its complexity.

LeetCode similar: LeetCode 135