1 round technical + 1 round behavioral. Technical round asked really detailed questions about your past experience and asked me to implement a 2 layer MLP using pytorch from scratch. Behavioral round was phone call held by HR, which lasts for 15 mins, basically just verified by education background, asked about my current location and desired compensation, etc.