Total 3 (or 4) rounds:
1. Basics - Python Coding (Easy DSA question), some PySpark, and a few statistics and maths related questions (AB testing, probability etc.)
2. ML Depth - classical ML algorithms, resume discussion, small DS case study
3. ML Breadth - resume discussion (initially), case study with ML system design
I was told that 3rd round might be the final round, but depending on the performance, there might be a 4th round as well. I never got any feedback or update from the recruiter after the third round, even after reaching out.