Lead Software Engineer, Data Platform
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 job
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
Go, Python, Svelte, TensorFlow, Tflite, Onnx, Mongodb Atlas, GCP, Azure, API
Similar jobs
ML Engineering jobs
Ontario, Canada
Design, build, and deploy production ML systems for recommendations, search, ranking, and advertising at internet scale. Own the full ML lifecycle from modeling to monitoring with strong cross-functional collaboration.
Dialpad
United States
Leads development of speech models, decoders, and low-latency inference systems for next-generation voice agents. Requires 5+ years in speech ML or related audio AI, strong Python and PyTorch experience, and the ability to guide technical direction and mentor engineers.
Ambience Healthcare
San Francisco, CA
Build and improve production AI systems for clinical products, owning evaluations, model behavior, agentic workflows, data flywheels, deployment, and observability. The role requires 5+ years of production ML or applied AI experience, strong Python and modern ML framework skills, and hands-on debugging expertise.
Fetch
United States
Build and operate low-latency machine learning systems for ad ranking, relevance, and optimization, including feature pipelines, experimentation, evaluation, and production inference. The role requires 6+ years of software engineering experience, strong Python skills, AWS experience, and practical LLM application experience.
Baselayer
San Francisco, CA
Senior AI Engineer responsible for production LLM agents that enrich business identity data through web discovery, verification, classification, and risk scoring. The role requires strong asynchronous Python, agent and evaluation expertise, browser automation, and experience operating AI systems in production.