Standard pytorch questions with some math and CS. Overall was not difficult if you have used pytorch regularly. Math questions on regression and classification. Some coding questions but they were more to test if you had a grasp of the language and not a leet code style round
Interview questions [1]
Question 1
How do you define a custom layer with a hand crafted gradient function
