Staff Machine Learning Engineer

Join our team at Workiva as a Staff Machine Learning Engineer! As a pivotal member of our Machine Learning (ML) team, you'll spearhead the architecture and delivery of groundbreaking machine learning solutions across our platform. Your expertise will be instrumental in leading projects that demand innovative problem-solving, including the integration of cutting-edge Generative AI into our products.In this role, you'll have the chance to develop robust tools, systems, and infrastructure to bolster the development, monitoring, and management of our machine learning solutions. Leveraging your engineering prowess, you'll tackle challenges related to availability and scaling, ensuring the long-term stability of our systems.If you're passionate about pioneering the possibilities of Generative AI and want to be part of a team driving innovation at Workiva, we invite you to join us! Learn more about Workiva's Generative AI and be part of shaping the future of ML with us.What You’ll DoArchitect and Develop SolutionsArchitect 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 and LeadCollaborate 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 requirementsEnsure Reliability and SupportDeliver, 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) What You’ll NeedRequired QualificationsBachelor’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 toolsetsPreferred QualificationsFamiliarity 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 performanceWorking ConditionsLess than 10% travelReliable internet access for remote working opportunities How You’ll Be Rewarded✅ Salary range in the US: $163,000.00 - $261,000.00✅ A discretionary bonus typically paid annually✅ Restricted Stock Units granted at time of hire✅ 401(k) match and comprehensive employee benefits packageThe salary range represents the low and high end of the salary range for this job in the US. Minimums and maximums may vary based on location. The actual salary offer will carefully consider a wide range of factors, including your skills, qualifications, experience and other relevant factors.Why Join WorkivaWorkiva is the platform designed to bring confidence, control, and a competitive edge to the world’s most complex organizations. Our AI-powered platform unifies finance, risk, and sustainability on a single, secure foundation—ensuring data is trusted, traceable, and ready to act on. With an unbroken path from source to output, leaders gain confidence in their numbers, visibility into current and emerging risks, and the ability to move with speed and precision in a constantly changing world.At Workiva, you’ll bring technology to market that executives, board