cover image
Strativ Group

Machine Learning Infrastructure Engineer (Lead/Staff)

On site

San francisco, United states

Senior

Full Time

19-01-2026

Share this job:

Skills

Research Training Machine Learning PyTorch TensorFlow OpenAI Robotics

Job Specifications

A stealth AI research lab building Physical AIis seeking a Machine Learning Infrastructure Lead to join its core team. The lab is already executing on commercial projects, with contracts in place and active collaboration across a highly technical group that includes alumni and contributors from MIT, Caltech, OpenAI, and leading industry labs. This is an opportunity to take ownership of the systems that power advanced AI models from training through real-world deployment.
The Machine Learning Infrastructure Lead will be responsible for designing, building, and scaling the lab's training and inference infrastructure. This is a senior, hands-on role focused on enabling rapid research iteration while delivering production-grade reliability for deployed systems interacting with the physical world.
You will work closely with research, robotics, and applied teams to translate cutting-edge models into robust, scalable systems. The ideal candidate has built and operated ML infrastructure in environments where performance, reliability, and scale are critical. Experience Required
4+ Years' exp. in SW/ML Engineering - strong focus on Infra & Systems
Significant hands-on experience with ML training and inference systems in production
Deep familiarity with PyTorch, JAX, or TensorFlow and large-scale training frameworks
Experience managing GPU clusters (cloud and/or on-prem), orchestration, and scheduling
Proven ability to partner closely with research teams to productionize models
This is an early-stage venture that will offer true ownership and growth. You will be building in a new, unsaturated space with vast potential.

About the Company

We are a multi award-winning team that does talent solutions. We live for innovation and work with ambitious businesses that make the world a better and safer place. Know more