First-round is general interaction(past projects, why would you join, etc.) with the Co-Founder of the company. The interviewer is obsessed with the amount of 'TensorFlow code' written in the last 6 months.
The second round is the case-study / coding challenge. The case study is again to check your TensorFlow skills. The case study is highly ambiguous with respect to its grammar, one should clarify on the expectations.