[aparvinash]

I might be wrong but

3 General White board coding rounds by SWEs(with some emphasis on data processing appropriate data structures, computations etc)

1 or 2 rounds specific to machine learning with some one working in that research area ( 1 or 2 based on the level you are coming in probably)

But as someone in the comments thread pointed out recruiter would be the best source