There was 1 research and 1 coding interview. The research interview consisted of 2 parts: in the first part, they asked questions about my publications, while the second part was a brainstorming session about a research problem.

1. ML design round: Asked a design question on ML compilers 2. LeetCode round: 2 easy-medium questions Was not able to go deep into compiler design as I was not aware that is what was required for the position(never mentioned in description)

I had two interviews: one for research fit, where I presented my own research and the interviewer asked a lot of open-ended questions and the second was 45 minutes of ML coding (no pytorch), which was not very hard.