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.

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.

there is one coding round, just two lc easy questions, and 1 research round, asked about my research papers, gave hypothetical situation and asked for specific model architecture design details. Overall was pretty normal and not hard.