Staff Software Engineer ML & AI