[Remote] Staff Machine Learning Engineer

Note: The job is a remote job and is open to candidates in USA. Workiva is a company focused on providing innovative solutions for complex organizations, and they are seeking a Staff Machine Learning Engineer to lead the architecture and delivery of machine learning solutions. The role involves developing robust tools and systems, collaborating with product teams, and ensuring the reliability of machine learning infrastructure.ResponsibilitiesArchitect and deliver cutting-edge ML solutions using MLOps and best practices, fostering creativity in project executionDesign systems to enable rapid ML development, high availability, and clear observabilityDevelop tools, systems, and automation to support ML solutions, ensuring efficiency, scalability, and rapid developmentCollaborate closely with product teams to develop APIs, maintain ML infrastructure, and integrate machine learning features into productsProvide technical leadership, mentor less experienced ML engineers and scientists, and define team best practices and processesLead in the ML space by introducing new technologies and techniques, and applying them to Workiva's strategic initiativesCommunicate complex technical issues to both technical and non-technical audiences effectivelyCollaborate with software, data architects, and product managers to design complete software products that meet a broad range of customer needs and requirementsDeliver, update, and maintain machine learning infrastructure to meet evolving needsHost ML models to product teams, monitor performance, and provide necessary supportWrite automated tests (unit, integration, functional, etc.) with ML solutions in mind to ensure robustness and reliabilityDebug and troubleshoot components across multiple service and application contexts, engaging with support teams to triage and resolve production issuesParticipate in on-call rotations, providing 24x7 support for all of Workiva’s SaaS hosted environmentsPerform Code Reviews within your group’s products, components, and solutions, involving external stakeholders (e.g., Security, Architecture)SkillsBachelor's degree in Computer Science, Engineering or equivalent combination of education and experienceMinimum of 4 years in ML engineering or related software engineering experienceProficiency in ML development cycles and toolsetsFamiliarity with Generative AIStrong technical leadership skills in an Agile/Sprint working environmentExperience building model deployment and data pipelines and/or CI/CD pipelines and infrastructureProficiency in Python, GO, Java, or relevant languages, with experience in Github, Docker, Kubernetes, and cloud servicesProven experience working with product teams to integrate machine learning features into the productExperience with commercial databases and HTTP/web protocolsKnowledge of systems performance tuning and load testing, and production-level testing best practicesExperience with Github or equivalent source control systemsExperience with Amazon Web Services (AWS) or other cloud service providersAbility to prioritize projects effectively and optimize system performanceBenefitsA discretionary bonus typically paid annuallyRestricted Stock Units granted at time of hire401(k) match and comprehensive employee benefits packageLess than 10% travelReliable internet access for remote working opportunitiesWorkiva supports employees in working where they work best - either from an office or remotely from any location within their country of employment.Company OverviewWorkiva created Wdesk, a cloud-based platform for enterprises to collect, manage, report, and analyze critical business data in real time. It was founded in 2008, and is headquartered in Ames, Iowa, USA, with a workforce of 1001-5000 employees. Its website is http://www.workiva.com.Company H1B SponsorshipWorkiva has a track record of offering H1B sponsorships, with 4 in 2025, 3 in 2024, 1 in 2023, 3 in 2022, 1 in 2021, 5 in 2020. Please note that this does not guarantee sponsorship for this specific role.