Four rounds.
1st round with recruiter, background checking.
2nd round is pytorch coding, modeling.
3rd round with hiring manager, go through a project accomplished by yourself.
4th round is a super round, 1 coding (leetcode), 1 knowledge deep dive (with CTO), 1 coding (pytorch), 1 research presentation.