Atoms
Posted 2mo ago

Senior Software Engineer - Machine Learning Infrastructure - San Francisco, CA

Atoms
San Francisco, California, United States
$176k-$230k/yrOnsiteFull Time
Responsibilities
  • building tools
  • maintaining infrastructure
  • researching innovations
Requirements
  • 3+ years software engineering experience
  • Background in data/ML/platform/ML infrastructure preferred
  • Proficiency in Python and Java
  • Familiarity with Azure, GCP, Docker, and Kubernetes
  • Ability to work independently and research ML innovations
Technical tools mentioned
PythonJavaAzureGCPDockerKubernetesLLM

Job description

Who we are 

Atoms is building the machines that power the next era of progress.

Over the last decade, software has transformed the digital world. But the physical world, where food is made, minerals are mined, goods are moved, and industries are run, remains far less intelligent, far less efficient, and far more constrained. We’re changing that.

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.

This work requires more than robotics. It requires deep integration across hardware, software, AI, operations, manufacturing, and real estate. We don’t just build machines in a lab. We deploy them into real environments, operate them, learn from them, and improve them until they work at scale.

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.

If you want to work on hard problems with real-world impact, join us.

 

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.

 

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.

 

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.

Actual compensation will be determined on an individual basis and may vary depending on experience, skills, and qualifications.

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

Benefits are subject to change at the company's discretion.
Atoms accepts applications on an ongoing basis.

Ready to join us as we serve those who serve others? 

#LI-Onsite

About Atoms

Private footwear making everyday sneakers for consumers.

Year founded
2018
Employees
32
Organization type
Private
Latest investment
Raised $2.12M Funding Round (2025)
Headquarters
US

Similar jobs

Software Engineer roles near San Francisco, California
1h
Save
Mark Applied
Hide
Software Engineer, Data Flywheel Platform
Sunnyvale or California or United States
OnsiteFull Time
Wayve
Wayve: British autonomous-driving software licensing vehicle-agnostic AI Driver technology to automakers and fleet owners.
6+ YOEProduction Python, large-scale data and distributed systems, SQL, system design, production operations, testing, observability, and on-call experience; approximately 6+ years of experience or equivalent practical expertise.
Python, Ray Data, Daft, Apache Spark, Databricks, turbopuffer, Milvus, Lance, Iceberg, Flyte, Apache Airflow, Dagster, PySpark, SQL, Kubernetes
11h
Save
Mark Applied
Hide
Software Engineer 2 - Inference
Vancouver or San Jose or Durham or Mexico City or Bangalore or Pune or Hoofddorp or Belgrade or Barcelona or Singapore or Sydney or Tokyo
$103k-$155k/yr HybridFull Time
Nutanix
NutanixNASDAQ: NTNX: Enterprise hybrid multi-cloud computing and software provider.
2+ YOE2–5 years developing resilient product software; strong data structures and algorithms; Docker, Kubernetes, Go, Python, CI/CD, distributed systems, datacenter architecture, and high-performance software experience; bachelor's or master's in computer science preferred.
Docker, Kubernetes, Go, Python, TensorFlow, PyTorch, CI/CD, Nutanix Cloud Platform
13h
Save
Mark Applied
Hide
Software Engineer II, Client Platform
San Mateo, California, United States
$163k-$245k/yr HybridFull Time
Sony Interactive Entertainment
Sony Interactive Entertainment: Global leader in interactive entertainment, hardware, and digital network services.
4+ YOEBachelor's degree or equivalent experience, 4+ years developing user-facing applications with JavaScript/TypeScript, React expertise, cross-platform experience, and familiarity with modern testing, APIs, CI/CD, containers, and AI tools.
JavaScript, TypeScript, React, Webpack, Babel, Jest, GraphQL, RESTful APIs, Docker, CI/CD, Apple
21h
Save
Mark Applied
Hide
Software Engineer - Game of Thrones Slots
Austin or San Mateo or Chicago or Toronto
$71k-$127k/yr HybridFull Time
Zynga
Zynga: Mobile video game developer and publisher creating games for players worldwide as a Take-Two subsidiary.
1+ YOEB.Sc. in Computer Science or equivalent; 1–2 years software development experience, or 3+ years relevant engineering experience; Unity, C#, PHP backend, Git, algorithms, and data structures.
Unity, C#, PHP, Git
22h
Save
Mark Applied
Hide
Software Engineer, GTM
San Francisco, California, United States
$160k-$200k/yr OnsiteFull Time
Hyperbound
Hyperbound: Private AI sales-coaching platform helping enterprise revenue teams practice conversations and improve performance.
Build and own production systems for lead routing, pipeline, forecasting, scoring, deal workflows, and revenue data across CRM, product, and billing systems.
23h
Save
Mark Applied
Hide
Senior Software Engineer, AI Ecosystem 
Mountain View, California, United States
$160k-$205k/yr OnsiteFull Time
Aerospike
Aerospike: Aerospike is a private software providing a real-time database platform for enterprise AI and mission-critical applications.
5+ YOERequires 5+ years coding experience in Go, Python, Java, Scala, C, or C++, plus distributed systems integration, Kubernetes, Docker, and DevOps/SRE expertise. NoSQL, vector databases, and open-source experience are bonuses.
Go, Python, Java, Scala, C, C++, Kubernetes, Docker, NoSQL, GitHub, Stack Overflow
1d
Save
Mark Applied
Hide
Software Engineer
San Francisco, California, United States
$150k-$300k/yr OnsiteFull Time
Arini
Arini: AI receptionist and healthcare automation software provider serving dental practices and dental service organizations.
Design, deploy, and iterate AI agents and automation workflows, integrate production code with enterprise systems, and deliver measurable customer outcomes in a fast-moving environment.
EHR
1d
Save
Mark Applied
Hide
Software Engineer
Menlo Park or Seattle
$219k-$301k/yr OnsiteFull Time
Meta
MetaNASDAQ: META: Builds technologies that help people connect, find communities, and grow businesses.
10+ YOEBachelor's degree or equivalent experience, 10+ years in networking or infrastructure software, 4+ years designing production dataplane/control-plane systems, Kubernetes networking, C/C++, scripting, and test automation.
Kubernetes, C, C++, Python, Shell Scripting, DPDK, eBPF/XDP, AF_XDP, io_uring, RDMA/RoCEv2, Linux, TC/iptables/nftables