Petuum Interview Question

Medium leetcode question and one ML question.