Staff Engineer, Machine Learning Platform