I took part in 3 rounds. Had I completed the 3rd round, there was one more round which was culture fit round. First was oncall round, tech screening + resume match. Second round was a data engineering round, SQL + 1 DSA questions(Minimum steps to make the product of the array equal to 1) + data engineering + SPARK + python Second round was with AI head : he made me implement vanilla version of airflow. Then a program to simulate throwing of 2 dices. and distribution of the sum.