Staff Software Engineer, ML Platform (SF, NYC, or Remote)