It took three rounds, and they asked for leet codes. Although I practiced a lot, I couldn't answer technical questions. My expectation was that because this is an ML position, they would ask about ML questions. Never make this mistake, even for research companies just ask leet code. As a researcher, it is not easy to learn all those codes in one week to write a new one in front of other Software developers in 10 mins.