Staff Machine Learning Infrastructure Engineer

What this role involves

Atoms is building Physical AI for industries like food, mining, and transport. The Staff Machine Learning Infrastructure Engineer will design and scale distributed GPU training infrastructure and MLOps pipelines to support autonomous transport models. This role requires deep experience with Kubernetes, distributed compute frameworks, and cross-functional collaboration with robotics and ML researchers.

Skills this posting asks for

• go
• python
• java
• rust
• kubernetes
• ray
• mlflow
• mlops
• distributed computing
• gpu training
• data engineering
• model validation
• experiment tracking

Requirements

• 8 years of experience
• Level: staff

From the employer’s posting

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 f…