Senior Software Engineer - Machine Learning Infrastructure
Tech Stack
Job Description, Responsibilities & Requirements
About the Position
Senior Software Engineer - Machine Learning Infrastructure
San Francisco, CA
About the Company
Atoms is building the machines that power the next era of progress. We are transforming the physical world by integrating hardware, software, AI, operations, manufacturing, and real estate to create Physical AI-real-world robots for industries like food, mining, and transport. Our mission is to make the physical world more intelligent, efficient, and scalable.
Responsibilities
- Work on the Machine Learning and AI Infrastructure team, focusing on providing infrastructure for data scientists and engineers to effectively use ML and AI technologies.
- Build and maintain software and tools for the classical ML stack, supporting the full training lifecycle from data collection, preparation, training, deployment, and model serving.
- Develop and maintain software and tools that enable efficient and effective use of GenAI (LLM) technologies.
- Conduct background research on the latest innovations in the Machine Learning space and apply insights within the team.
Requirements
- 3+ years of experience as a software engineer, with the ability to work independently in a fast-paced, dynamic environment.
- Background in data infrastructure, platform infrastructure, or ML infrastructure is preferred.
- Proficiency in Python and Java programming languages, with experience in 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.
We Offer
- Competitive base salary range: $176,000 - $230,000 per year.
- Equity awards and annual performance-based bonus.
- Comprehensive benefits package including:
- 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).
- Generous Time Off policies.
About the Company
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.
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.
Ready to join us as we serve those who serve others?
#LI-Onsite