First stage: probability questions, linear algebra, and 1-2 pure math questions. Second stage: Interview. During the interview I was also encouraged to mentally calculate approximate solutions to the probability questions.