Machine Learning on the Edge for 5G