Staff Software Engineer - Machine Learning

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 Do

Architect and Develop Solutions

• Architect and deliver cutting-edge ML solutions using MLOps and best practices, fostering creativity in project execution

• Design systems to enable rapid ML development, high availability, and clear observability

• Develop tools, systems, and automation to support ML solutions, ensuring efficiency, scalability, and rapid development

Collaborate and Lead

• Collaborate closely with product teams to develop APIs, maintain ML infrastructure, and integrate Machine Learning features into products

• Provide technical leadership, mentor less experienced ML engineers and scientists, and define team best practices and processes

• Lead in the ML space by introducing new technologies and techniques, and applying them to Workiva's strategic initiatives

• Communicate complex technical issues to both technical and non-technical audiences effectively

• Collaborate with software, data architects, and product managers to design complete software products that meet a broad range of customer needs and requirements

Ensure Reliability and Support

• Deliver, update, and maintain Machine Learning infrastructure to meet evolving needs

• Host ML models to product teams, monitor performance, and provide necessary support

• Write automated tests (unit, integration, functional, etc.) with ML solutions in mind to ensure robustness and reliability

• Debug and troubleshoot components across multiple service and application contexts, engaging with support teams to triage and resolve production issues

• Participate in on-call rotations, providing 24x7 support for all of Workiva’s SaaS hosted environments

• Perform Code Reviews within your group’s products, components, and solutions, involving external stakeholders (e.g., Security, Architecture) 

What You’ll Need

Required Qualifications

• Bachelor’s degree in Computer Science, Engineering or equivalent combination of education and experience

• Minimum of 4 years in ML engineering or related Software Engineering experience

• Proficiency in ML development cycles and toolsets

Preferred Qualifications

• Familiarity with Generative AI

• Strong technical leadership skills in an Agile/Sprint working environment

• Experience building model deployment and data pipelines and/or CI/CD pipelines and infrastructure

• Proficiency in Python, GO, Java, or relevant languages, with experience in Github, Docker, Kubernetes, and cloud services

• Proven experience working with product teams to integrate Machine Learning features into the product

• Experience with commercial databases and HTTP/web protocols

• Knowledge of systems performance tuning and load testing, and production-level testing best practices

• Experience with Github or equivalent source control systems

• Experience with Amazon Web Services (AWS) or other cloud service providers

• Ability to prioritize projects effectively and optimize system performance

Working Conditions

• Less than 10% travel

• Reliable internet access for remote working opportunities

How You’ll Be Rewarded

✅ Salary range in the US: $120,000.00 - $204,000.00

✅ A discretionary bonus typically paid annually

✅ Restricted Stock Units granted at time of hire

✅ 401(k) match and comprehensive employee benefits package

The 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.

Where You’ll Work

Our values drive how we work and who we hire. You will see these values ingrained in how we support our customers, work with team members, build our products and in the work environment we’ve created.

We believe our people are our greatest asset, and our unique culture gives employees the opportunity to make an impact everyday. We give our employees the freedom and resources they need—backed by our culture of collaboration and diverse thought—to continue innovating and breaking new ground. We hire talented people with a wide range of skills and experiences who are eager to tackle some of today’s most challenging problems.

At Workiva, you’ll enjoy

• Fantastic Benefits: With coverage starting day one, choose from competitive health, dental, and vision plans on the largest physician networks available.

• Casual Dress: Workiva has a casual work environment, most people wear jeans to the office.

• Involvement: Ability to participate in Business Employee Resource Groups (Black, Hispanic, Asian, Women, Rainbow (LGBTQIA+), Veterans, Disabilities), Volunteering, Company wide celebrations, and more

• Work-life Balance: We have competitive PTO, VTO and Parental Leave. We encourage employees to spend time enjoying life outside of work.  

Learn more about life at Workiva: https://www.linkedin.com/company/workiva/

Learn more about benefits: https://www.workiva.com/careers/benefits  

Workiva is an Equal Employment Opportunity and Affirmative Action Employer.  We believe that great minds think differently.  We value diversity of backgrounds, beliefs, and interests, and we recognize diversity as an important source of intellectual thought, varied perspective, and innovation.  Employment decisions are made without regard to age, race, creed, color, religion, sex, national origin, ancestry, disability status, veteran status, sexual orientation, gender identity or expression genetic information, marital status, citizenship status or any other protected characteristic.  We strongly encourage and welcome people from historically marginalized groups to apply.

Workiva is committed to working with and providing reasonable accommodations to applicants with disabilities. To request assistance with the application process, please email talentacquisition@workiva.com.  

Workiva employees are required to undergo comprehensive security and privacy training tailored to their roles, ensuring adherence to company policies and regulatory standards.

Workiva supports employees in working where they work best - either from an office or remotely from any location within their country of employment.

#LI-HS1