There were 3 rounds in totals. The first round was with the manager. He will go test you on some parallel topics, a puzzle to solve, and a question about C++ programming language. The two other rounds are about Machine Learning (simple questions about autograd, parallel inference in CPU) and Technical (coding in general). Then you will have 2 assignments.