← Back to results

Senior Software Engineer - Machine Learning Infrastructure

Join Atoms as a Senior Software Engineer to build and maintain ML infrastructure for real-world applications in a dynamic environment.

Location
San Francisco, CA
Compensation
$176k–$230k/yr
Level
senior
Type
full time · On-site

Posted by employer 4 months ago

First seen on Joblaze 1 month ago

Last verified on the company career page 1 day ago

Apply at Atoms → Save job Scanned from atoms.co

Skills & Technologies

AI in the day-to-day

Build and maintain software and tools for classical ML stack and GenAI technologies.

Requirements

Experience
3+ years

Not disclosed in this posting: visa sponsorship.

Benefits

401k Match Equity/Stock Options Health Insurance Parental Leave

Joblaze summary

In this role, the Senior Software Engineer will focus on developing and maintaining the machine learning infrastructure that supports data scientists and engineers in their work with AI technologies. Key skills include proficiency in Python and Java, along with experience in cloud environments like Azure and GCP, as well as familiarity with deployment tools such as Docker and Kubernetes. This position is ideal for someone with over three years of experience in software engineering, particularly in data or ML infrastructure, who thrives in a fast-paced environment. Atoms emphasizes collaboration and innovation, making it a fitting place for those passionate about solving complex real-world pr

Joblaze insights

  • Listed about a month ago — first seen on Joblaze September 3, 2026. Last confirmed on Atoms's careers page October 31, 2026.
  • Salary band is in line with the typical range for AI/ML roles (median ~$190,000).
  • Starts above 36% of 206 comparable senior ai/ml roles in United States that list Python we track (median $182,000 across 71 companies). See Python salary trends
  • Python appears in 53.1% of 556 comparable senior ai/ml roles in United States; Docker appears in 4.7% of 556 comparable senior ai/ml roles in United States.

Quick facts

Is the Senior Software Engineer - Machine Learning Infrastructure role remote?
No — this is an on-site role in San Francisco, CA.
What's the salary range?
Atoms lists $176,000–$230,000 for this role.
How much experience is required?
At least 3 years of relevant experience for this Senior Software Engineer - Machine Learning Infrastructure role.
Where is the role based?
Atoms is hiring for this position in San Francisco, CA.
What's the tech stack?
Joblaze extracted these technologies from the posting: Azure, Docker, GCP, Java, Kubernetes, Python.
What seniority level is this role?
Atoms targets senior candidates for this position.
Is this full-time or contract?
Full-time for this Senior Software Engineer - Machine Learning Infrastructure role at Atoms.

From the original posting

Atoms builds Physical AI— real-world robots for the industries that move civilization forward, starting with food, mining, and transport. Our systems are designed to understand, predict, and control the real world with precision, turning complex physical operations into something more reliable, more scalable, and more productive.

We are roboticists, engineers, operators, and builders. We believe the next great technology companies will not only transform information, but the physical systems that shape everyday life.

What you’ll do

  • Work on the Machine Learning and AI Infrastructure team, which is focused on providing infrastructure for data scientists and other engineers to use ML and AI technologies effectively.
  • Build and maintain software and tools for classical ML stack which helps data scientists manage full training lifecycle from data collection, data preparation, training, deployment and model serving.
  • Build and maintain software and tools which enable efficient and effective use of GenAI (LLM) technologies.
  • Do background research on latest innovations in the Machine Learning space and bring back insights to apply within the team.

What we’re looking for

  • 3+ years of experience as a software engineer, and the ability to work independently in a fast-paced, dynamic environment, taking initiative to drive results.
  • Background in data infrastructure, platform infrastructure or ML infrastructure is preferred.
  • Enthusiastic about learning new codebases, and making changes in large complex systems safely and efficiently.
  • Proficiency in Python and Java programming languages. We’re looking for engineers who are comfortable with multiple languages and have experience building scalable high-performance systems
  • Familiarity with Azure and GCP cloud environments, and deployment tools such as Docker and Kubernetes
  • Experience in real estate, food technology, or related industries is a bonus, though not required. A passion for solving complex problems in these sectors is key.

What else you need to know

This role is based in our San Francisco office location. As a company driven by innovation and continuous change, close collaboration is essential. We’re constantly reimagining our industry, creating new products, and refining our processes, and we do our best work together. That’s why all of our office-based teams work onsite, five days a week.

The base salary range for this role is $176,000 - $230,000 per year.

Base salary is just one part of your total rewards package. You may also be eligible for equity awards and an annual performance-based bonus.

Benefits Summary (USA Full-Time Exempt Employees):

  • Medical, dental, and vision insurance (multiple plans, incl. HSA options).
  • Company-paid life and disability insurance (short- and long-term).
  • Voluntary insurance: accident, critical illness, hospital indemnity.
  • Optional supplemental life insurance for self, spouse, and children.
  • Pet insurance discount.
  • 401(k).
  • Health Savings Account (HSA)
  • Flexible Spending Accounts (Healthcare, Dependent Care, Commuter)
  • Time Off policies:
    • Discretionary vacation days
    • 8 paid holidays per year
    • Paid sick time
    • Paid Bereavement leave
    • Paid Parental Leave

#LI-Onsite

Standard company text repeated across Atoms's postings is omitted here.

Similar positions

Atoms
Senior Machine Learning Engineer
Atoms · Mountain View, CA
Atoms
Atoms
Senior Full Stack Engineer
Atoms · San Francisco, CA
Atoms
Atoms
Senior Machine Learning Engineer
Atoms · San Francisco, CA