Senior/Staff Software Engineer - ML Infrastructure