Skip to content
ViamViamNew York, NY

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.

220k – 250k/yr
HybridML Engineering

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 AtlasGCPAzureAPI

Similar roles

ML Engineering jobs
Attentive

Senior Software Engineer, ML/AI Platform

AttentiveUnited States

Senior Software Engineer on the ML Platform team building and operating data, tooling, serving, and inference layers for a PB-scale feature store and ML lifecycle at a high-growth AI marketing company.

220k – 260k/yr
Remote5+ YOEML Engineering
Discord

Senior Software Engineer, Machine Learning (Ads)

DiscordSan Francisco, CA

Develops and deploys ML models for ads targeting, ranking, and measurement on Discord's Ads platform. Requires 5+ years ML experience, 3+ years in Ads ML, Python proficiency, and PyTorch/TensorFlow expertise.

220k – 247k/yr
On-site5+ YOEML Engineering
Bluesky Social

Senior Machine Learning Engineer

Bluesky SocialSeattle, WA

Designs, develops, and maintains ML systems for content recommendations, search, spam detection, and labeling using Bluesky's social graph. Requires 3+ years in ML/data science focused on recommendations/search, Python/PyTorch proficiency, and rapid experimentation skills.

221k – 405k/yr
Remote3+ YOEML Engineering
Reddit

Senior Machine Learning Engineer, Ads Content Understanding

RedditUnited States

Senior Machine Learning Engineer building and shipping production ML systems for Ads Content Understanding at Reddit. Focus on LLMs, content taxonomies, opinion mining, product understanding, and operationalizing signals for monetization impact. Requires 5+ years delivering scalable ML in production.

217k – 303k/yr
Remote5+ YOEML Engineering
Reddit

Senior Machine Learning Systems Engineer

RedditUnited States

Build large-scale ML experimentation and training orchestration platforms, including agentic AI execution systems, to accelerate Ads ML development at Reddit. Requires 5+ years infrastructure experience and 2+ years building production ML platforms.

217k – 303k/yr
Remote5+ YOEML Engineering