Staff Software Engineer, AI/ML