Staff Software Engineer, ML Platform