Skip to content

Lead Software Engineer, Data Platform

220k – 250kNew York, NYML EngineeringHybrid
Summary

Viam is seeking a Lead Software Engineer to lead the Data/ML team, owning technical direction, architecture, and delivery. This hands-on role involves managing a team of 5+ engineers, writing code, and driving the reliability and performance of ML training and inference infrastructure.

About the role

About the Role

We are looking for a Lead Engineer to own the technical direction and delivery of our Data/ML function. This is a hands-on leadership role where you will manage a team, make architectural decisions, and code. The Data/ML team owns the infrastructure that moves data from devices to the cloud and makes it usable across the Viam platform. Performance, reliability, usability, and correctness affect every solution built on Viam. On top of that foundation, the team also owns the ML infrastructure that turns that data into models: training workflows, labeling pipelines, and inference in both cloud and at the edge. Some of the interesting challenges they solve: handling an unusually wide variety of data types from devices that are frequently offline, building auto-labeling and training workflows at scale, and expanding cloud inference across multiple model backends. Tech stack is Go and Python with a Svelte frontend, running ML workloads on GKE using TensorFlow, TFLite, and ONNX, with MongoDB Atlas, GCP, and Azure for the data layer. You will report to the VP of Engineering.

What You'll Own

  • Lead and develop a team of 5+ engineers: set direction, run planning, ship reliably, and grow the team through coaching, feedback, and performance conversations
  • Write code and ship features alongside your team across the full stack, from backend infrastructure through UI
  • Own the architecture of the data pipeline end to end, from device to cloud, including storage, querying, and the APIs that power the rest of the platform
  • Drive the reliability and performance of ML training and inference infrastructure, from custom training workflows to the services that power auto-labeling and model evaluation
  • Have opinions on what gets built and how users experience it, and bring those into product decisions early
  • Own cross-team problems through to resolution, working closely with Computer Vision, Fleet Management, Mobile, and Solutions Engineering

What You Bring

We are looking for a technical lead with strong systems judgment and product instincts who can set direction and stay hands-on.

  • You have led engineers and know how to develop them, not just manage them
  • You have a background building and scaling backend or platform infrastructure in production
  • You have shipped APIs and UIs and think about both as part of the same product problem
  • You have sound judgment on system design and tradeoffs under real constraints
  • You have product instincts and use them: when something is being built the wrong way, you say so early
  • You make decisions with incomplete information and have a bias toward shipping over planning
  • You have a track record of driving cross-team projects from ambiguity to production
  • Experience with data-intensive systems, streaming pipelines, or ML infrastructure

Full-time Benefits

  • Viam’s base salary range for this role is posted above. In addition to cash compensation, Viam offers a comprehensive Total Rewards package that includes equity grants, health benefits, and more.
  • 100% covered medical/dental/vision insurance plans, commuter benefits
  • 25 days paid vacation and generous holiday observances
  • Free lunch every day that you’re in the office
  • One Medical Membership
  • Citi Bike memberships
  • Monthly wellness stipend to be used for a variety of fitness-related items like gym memberships, fitness classes, fitness equipment, and more
  • Paid parental leave
  • Reproductive Health Benefits, including Fertility Benefits and Abortion Access Travel Benefits
  • Yearly Learning and Development Budget for your own personal learning goals
Skills
GoPythonSvelteTensorFlowTFLiteONNXMongoDB AtlasGoogle Cloud PlatformAzureAPI
Similar roles at this salary range
All ML Engineering jobs →
Ironclad

Senior Software Engineer, AI

Lead design and delivery of high-priority AI initiatives across multiple codebases. Build and ship AI-powered features with strong backend fundamentals and product sense.

180k – 220kSan Francisco, CAML EngineeringHybrid5+ YOEReactEvals
Plaid

Machine Learning Engineer - Embedded Insights

Drive ML initiatives from concept to production on the Embedded Insights team. Identify opportunities, build and deploy models using Plaid's financial datasets, and partner with product teams to deliver scalable customer-facing intelligence products.

212k – 272kSan Francisco, CA +2ML EngineeringHybrid5+ YOESQLMLOps
Plaid

Machine Learning Engineer

Advance Plaid’s foundation models by developing novel architectures, pretraining objectives, and fine-tuning strategies. Work across the full ML stack from data engineering to production serving and monitoring.

212k – 272kSan Francisco, CA +2ML EngineeringHybrid1+ YOELLMsPython
Airbnb

Senior Machine Learning Engineer

Build and deploy cutting-edge Agentic AI and LLM systems to transform Airbnb's customer service experience, including Chat and Voice AI assistants. Requires 6+ years experience with production ML/AI systems at scale.

196k – 227kUnited StatesML EngineeringRemote6+ YOELLMSFT
Decagon

Staff Software Engineer, Agents

Build and own end-to-end AI agents for enterprise customers, integrating latest text/voice models and iterating based on real-world usage. Requires 8+ years of software engineering experience with Python and TypeScript.

200k – 400kSan Francisco, CAML EngineeringOn-site8+ YOEPythonAI Agents