Implement K-Means in NumPy or PyTorch

Problem Overview

This is a recurring Apple ML Coding question for Machine Learning Engineer candidates (phone screen and onsite, on a HackerRank-style shared editor). You are asked to implement K-Means clustering...