Apart from the rounds with the hiring manager and HR, all the rounds were focused on data structure and algorithm problems from Leetcode (Medium and Hard). If someone is prepared thoroughly, he should be able to answer most of the questions. The other type of questions were on machine learning fundamentals, models, and open-ended questions of machine learning system design. The manager and HR did ask behavioral questions related to team work, working in agile methodology and also how to tackle difficult challenges.