Google DeepMind interview question

More mathy/ML leetcode style coding