Argo AI Machine Learning Engineer interview questions

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