Senior Software Engineer - Machine Learning Infrastructure

About the role

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.

What You'll Do

• 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.

Why join us

At Atoms, you'll work on one of the defining challenges of our time—bringing automation into the physical world to drive real, lasting impact. We exist to uncover valuable unknown truths and turn them into progress, which means constantly pushing beyond what's known and building what doesn't yet exist. The work is ambitious and often challenging, but it's grounded in a shared sense of purpose and a team committed to seeing it through together. Our work only matters if it serves others, and we know that meaningful progress depends on the trust of the people we serve and the strength of our team—so we invest in both, creating an environment where you can do your best work and grow.

Pay

The base salary range for this role is $176,000 - $230,000 per year. You may also be eligible for equity awards and an annual performance-based bonus.

Schedule

This role is based in our San Francisco office location. All office-based teams work onsite, five days a week.

Benefits

• 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)
• Discretionary vacation days
• 8 paid holidays per year
• Paid sick time
• Paid Bereavement leave
• Paid Parental Leave