HR round - Self introduction, introduction to company and explanation of interview procedure. Round1 - ML basics and programming in pytorch (I don't remember the exact question) but it was simple algorithm to implement. All the interviewer wanted to check was how profienct I was in using the coding language. Round2- Computer vision concepts.