Senior Staff Machine Learning Engineer
Location
United States
Posted
2 days ago
Salary
$193K - $308K / year
Seniority
Senior
Job Description
Senior Staff Machine Learning Engineer
Workiva
• Own the architecture of Workiva’s AI platform and core AI services • Shape how machine learning, Generative AI, and agentic systems are integrated across products • Lead the move from early adoption to production-grade, enterprise-ready systems • Define standards for model serving, retrieval, evaluation, governance, and platform reliability • Lead the design of enterprise agentic systems, including orchestration, workflow execution, memory, and multi-agent coordination • Design and evolve Retrieval-Augmented Generation capabilities for enterprise content and knowledge workflows • Establish evaluation methods and quality frameworks for Generative AI applications • Assess emerging AI technologies and guide adoption strategy for Workiva’s platform • Influence technical direction across teams, products, and platform domains • Mentor Staff and Senior Engineers and help raise the technical bar across the organization • Partner closely with Product, Security, Infrastructure, and Architecture leaders • Align teams around a shared vision for scalable, secure AI at Workiva • Lead secure AI platform design, including authorization, runtime isolation, governance, auditability, and compliance • Establish best practices for AI safety, model governance, and customer data protection • Ensure AI systems meet enterprise expectations for availability, resiliency, observability, and operational support • Design for fault tolerance and operational excellence in regulated, security-conscious environments
Job Requirements
- Bachelor’s degree in Computer Science, Engineering, or equivalent experience
- 10+ years of software engineering experience, including large-scale SaaS platforms
- 5+ years designing, deploying, and operating production ML, AI, or data-intensive systems
- Experience designing and operating enterprise AI platforms, including model serving, evaluation, observability, and governance.
- Deep expertise in RAG, agentic systems, and large-scale knowledge systems
- Strong understanding of foundation model ecosystems, including inference, routing, prompting, and provider tradeoffs
- Experience with AI evaluation, secure AI systems, and regulated enterprise environments
- Proven track record leading architecture across multiple teams or platform domains
- Strong distributed systems, cloud-native, API, reliability, and operational excellence experience
- Expert-level Python proficiency and proficiency in at least one production language such as Java, Go, Scala, or C++
- Proven record mentoring senior engineers and technical leaders.
Benefits
- A discretionary bonus typically paid annually
- Restricted Stock Units granted at time of hire
- 401(k) match and comprehensive employee benefits package
Related Guides
Related Job Pages
More Machine Learning Engineer Jobs
• Evolve and maintain our Kubeflow, Feast and Spark-on-Kubernetes infrastructure, ensuring it can handle the increasing complexity of both traditional ML and the new wave of AI. • Design the internal tools, APIs, and abstractions that empower distributed teams to own their entire lifecycle—transitioning our data culture from "centralised bottleneck" to "self-service excellence." • Collaborate with embedded Data Science teams to adapt software engineering best practices (CI/CD, versioning, testing) to ML-specific workflows, raising the bar for production-grade AI across the company. • Drive MLOps best practices and define the specific lifecycle requirements for LLMOps, ensuring a frictionless journey from experimental notebooks to robust, production-grade solutions by providing top-notch deployment frameworks and automation. • Partner closely with our Infrastructure and Data squads to break silos and integrate ML artifacts into our global Data Catalog, Privacy, and Governance frameworks. • Inspire and empower others by genuinely caring for your own wellbeing and your colleagues.
• Design, develop, and maintain scalable data pipelines to support ingestion, transformation, and delivery into centralized feature stores, model-training workflows, and real-time inference services. • Build and optimize workflows for extracting, storing, and retrieving semantic representations of unstructured data to enable advanced search and retrieval patterns. • Architect and implement lightweight analytics and dashboarding solutions that deliver natural language query experience and AI-backed insights. • Define and execute processes for managing prompt engineering techniques, orchestration flows, and model fine-tuning routines to power conversational interfaces. • Oversee vector data stores and develop efficient indexing methodologies to support retrieval-augmented generation (RAG) workflows. • Partner with data stakeholders to gather requirements for language-model initiatives and translate into scalable solutions. • Create and maintain comprehensive documentation for all data processes, workflows and model deployment routines. • Should be willing to stay informed and learn emerging methodologies in data engineering, MLOps and LLM operations.
• Debug and troubleshoot complex deployment and integration challenges across customer environments • Work directly with customer engineering and platform teams to deploy and operationalize AI systems • Design and implement integrations across enterprise AI workflows, APIs, infrastructure, and governance platforms • Deploy and support AI evaluation frameworks, observability solutions, and guardrail implementations • Translate customer deployment challenges into actionable feedback for product and engineering teams • Partner with engineering, infrastructure, security, risk, compliance, and operations stakeholders to navigate enterprise governance requirements • Drive successful customer adoption of AI infrastructure solutions while balancing technical and business requirements
Senior Machine Learning Engineer - Platform
SamsaraSamsara Inc. is on a mission to increase the sustainability of the operations that power the global economy. The company pioneers the Connected Operations Cloud
Who we are Samsara (NYSE: IOT) is the pioneer of the Connected Operations™ Cloud, which is a platform that enables organizations that depend on physical operations to harness Internet of Things (IoT) data to develop actionable insights and improve their operations. At Samsara, we are helping improve the safety, efficiency and sustainability of the physical operations that power our global economy. Representing more than 40% of global GDP, these industries are the infrastructure of our planet, including agriculture, construction, field services, transportation, and manufacturing — and we are excited to help digitally transform their operations at scale. Working at Samsara means you’ll help define the future of physical operations and be on a team that’s shaping an exciting array of product solutions, including Video-Based Safety, Vehicle Telematics, Apps and Driver Workflows, and Equipment Monitoring. As part of a recently public company, you’ll have the autonomy and support to make an impact as we build for the long term. About the role: We are looking for a Senior Machine Learning Engineer to lead the architectural evolution of our safety systems. You will move our ML stack from siloed, end-to-end models toward a unified Perception Platform Layer. Your mission is to build the robust infrastructure that translates raw sensor data into real-time, high-stakes decisions, ensuring our models perform reliably across both cloud and edge environments. This is a remote position for candidates based in the US. You should apply if: - You want to impact the industries that run our world: The software, firmware, and hardware you build will result in real-world impact—helping to keep the lights on, get food into grocery stores, and most importantly, ensure workers return home safely. - You want to build for scale: With over 2.3 million IoT devices deployed to our global customers, you will work on a range of new and mature technologies driving scalable innovation for customers across industries driving the world's physical operations. - You are a life-long learner: We have ambitious goals. Every Samsarian has a growth mindset as we work with a wide range of technologies, challenges, and customers that push us to learn on the go. - You believe customers are more than a number: Samsara engineers enjoy a rare closeness to the end user and you will have the opportunity to participate in customer interviews, collaborate with customer success and product managers, and use metrics to ensure our work is translating into better customer outcomes. - You are a team player: Working on our Samsara Engineering teams requires a mix of independent effort and collaboration. Motivated by our mission, we’re all racing toward our connected operations vision, and we intend to win—together. In this role, you will work on the following: 1. Platform Architecture & Unification - Architect a Unified Perception Layer: Lead the transition from fragmented, task-specific models to a modular perception platform that supports reusable components and downstream safety applications. - System Design: Design and implement real-time ML systems—from sensor ingestion and tracking to risk reasoning and actuation—ensuring clear interfaces and predictable system behavior. - Hybrid Deployment: Orchestrate model integration across edge and cloud environments, managing versioning, rollouts, and mission-critical fallback mechanisms. 2. Performance & Reliability Engineering - Latency Ownership: Own end-to-end latency and reliability for safety-critical pipelines. You will profile, schedule, and optimize messaging and backpressure across the entire stack. - Observability & Feedback Loops: Build sophisticated monitoring for deployed models to detect drift, false positives/negatives, and latency regressions. You will "close the loop" to ensure production data informs the next iteration of training. 3. Rigorous Evaluation & Safety - Safety Cases: Develop evaluation frameworks specifically for rare "long-tail" safety events. You will define metrics and build targeted test sets that form the basis for principled ship/no-ship decisions. - Explainability: Partner with Applied Scientists to ensure research outputs are translated into production code that is not only performant but also debuggable and explainable. 4. Technical Leadership - Strategic Influence: Shape the system abstractions early in the platform transition to minimize technical debt and maximize future scalability. - Mentorship: Set the engineering standard for correctness and performance. You will mentor junior and mid-level engineers, fostering a culture of rigorous ML engineering. Minimum requirements for the role: - Experience: 6+ years of experience in ML Engineering, with a proven track record of shipping models in production (ideally in safety-critical domains like robotics, automotive, or industrial AI). - Systems Mastery: Deep understanding of distributed systems, performance profiling, and computer vision. - Infrastructure Fluency: Experience with Cloud ML workflows (AWS/GCP/Azure) and containerization, paired with an understanding of the constraints of edge hardware. - Architectural Mindset: You don't just write code; you design systems. You understand the trade-offs between model complexity and operational reliability. An ideal candidate also has: - Ph.D. in Computer Science or quantitative discipline (e.g., Applied Math, Physics, Statistics) - Experience with containerization technologies (e.g., Docker, Kubernetes), continuous integration/continuous deployment (CI/CD) pipelines, and infrastructure-as-code (IaC) frameworks - Familiar with deploying and managing ML applications in cloud environments, as well as leveraging cloud-based services for data storage, processing, and inference - Experience building end-to-end ML applications from scratch The range of annual base salary for full-time employees for this position is below. Please note that base pay offered may vary depending on factors including your city of residence, job-related knowledge, skills, and experience. This role is also eligible for an initial RSU grant with no vesting cliff, and ongoing refresh opportunities tied to performance, subject to plan terms and conditions. Learn more about our total rewards and benefits below. Annual Base Salary $157,250—$203,500 CAD Total Rewards At Samsara, we build for the people who keep the global economy moving. We want owners, not passengers, which is why our rewards are designed to fuel high-impact builders. Our compensation program delivers above-market total compensation through a combination of base salary, performance-based bonus/variable pay, and equity (for eligible roles) in a high-growth public company. We meaningfully differentiate pay for our top performers, who have the opportunity to earn above-market compensation that can outpace the broader market over time. Beyond compensation, we provide the foundations that enable long-term success: a flexible, employee-led remote model, a professional development stipend, comprehensive health and parental leave plans, and more. If you’re ready to build for the long term and own the outcome, your journey starts here. Flexible Working At Samsara, we embrace a flexible working model that caters to the diverse needs of our teams. Our offices are open for those who prefer to work in-person and we also support remote work where it aligns with our operational requirements. For certain positions, being close to one of our offices or within a specific geographic area is important to facilitate collaboration, access to resources, or alignment with our service regions. In these cases, the job description will clearly indicate any working location requirements. Our goal is to ensure that all members of our team can contribute effectively, whether they are working on-site, in a hybrid model, or fully remotely. All offers of employment are contingent upon an individual’s ability to secure and maintain the legal right to work at the company and in the specified work location, if applicable. Belonging at Samsara At Samsara, we welcome everyone regardless of their background. All qualified applicants will receive consideration for employment without regard to race, color, religion, national origin, sex, gender, gender identity, sexual orientation, protected veteran status, disability, age, and other characteristics protected by law. We depend on the unique approaches of our team members to help us solve complex problems and want to ensure that Samsara is a place where people from all backgrounds can make an impact. Accommodations Samsara is an inclusive work environment, and we are committed to ensuring equal opportunity in employment for qualified persons with disabilities. Please email accessibleinterviewing@samsara.com or click here if you require any reasonable accommodations throughout the recruiting process. Our Commitment to Authenticity We use Tofu, a fraud detection tool, to validate the authenticity of applications and protect against identity fraud. This ensures we are connecting with real people and allows us to prioritize genuine candidates. Please see Samsara’s Candidate Privacy Notice for more information. Fraudulent Employment Offers Samsara is aware of scams involving fake job interviews and offers. Please know we do not charge fees to applicants at any stage of the hiring process. Official communication about your application will only come from emails ending in @samsara.com, @us-greenhouse-mail.io or @mail3.guide.co. For more information regarding fraudulent employment offers, please visit our blog post here.




