Round one - 20 questions on linear algebra, optimization, neural networks. It was easy for someone how has a background in math or CS. There were some questions on PCA, effect of different activation functions and etc.

There was a technical screening test, then a 30 min interview to check how the thought process in the test. The next round was with a senior manager who asked an open ended probability question and questions related to parameters in cnn and fully connected network