interview consisted one phone screen followed by 7 interviews covering Python, Cuda Programming, Pytorch/Tensorflow and ML techniques like transformers vs CNNs.
The phone screening consisted of a lead code problem related to graph traversal.
Python questions were related to decorators and containers. For Cuda Programming was asked to parallelise K-Means clustering algorithm. For system design I was asked to design a augmentation framework