ML Infrastructure Engineer
Gritt Robotics
South San Francisco, CA · Posted May 26, 2026
Just laid off? Start here: a plan for the first weeks
See every open role at Gritt Robotics
More in California job market · Technology & Software · Data & Analytics jobs
More roles at Gritt Robotics
- Field Software Engineer · South San Francisco, CA
- Robotics Software - Perception ML Engineer · South San Francisco, CA
- Robotics System Test Engineer · South San Francisco, CA
- Mechanical Engineer · South San Francisco, CA
- Robotics Software - Planning & Controls Engineer · South San Francisco, CA
- Construction Manager
Similar jobs
- #Machine Learning Engineer - Generative AI · Qualcomm Technologies · San Diego, CA
- Developer Relations Manager, Higher Education and Research - Foundational AI · Nvidia · Santa Clara, CA
- Computer Scientist (Ph.D.) · Exponent · Menlo Park, CA
- Electrical Engineer/Scientist/Computer Scientist (Ph.D.) · Exponent · Los Angeles, CA
- Principal Forecasting and Planning - AI Annotation Program · Intuit · San Diego, CA
- Staff People Analyst - Applied AI · Intuit · Mountain View, CA
Data & Analytics jobs in the rankings
Rankings about data & analytics jobs
- Best cities for data and analytics jobs #1 New York (1,713 openings)
- Best cities for data and analytics pay after cost of living #1 Foster ($232k)
- Most requested AI skills in data and analytics jobs #1 Machine learning engineering (17.1%)
- AI tools named most often in data and analytics jobs #1 Databricks (7.6%)
- Employers with the most data and analytics jobs #1 Amazon (711 roles)
Data & Analytics jobs · Top Data & Analytics employers · All rankings
How AI is used in this kind of work
Among computer and information research scientists, AI is already used for a large share of the tasks. Measured exposure for computer and information research scientists: 34% (high).
Share of this job’s tasks that AI is already used for at work, from real usage. It is not a forecast of job loss, and a low number means no use seen yet. Source: Anthropic Economic Index.
Skills this job asks for
Say which you have and which you want to build. We use it to match you to jobs like this one.
About the job
Gritt https://www.gritt.ai/ is an intelligent system that combines robotics and AI to build the infrastructure that pulls society forward. Gritt deploys via simple attachments to common equipment found on construction sites and autonomously performs labor-intensive tasks, verification, and planning. Gritt systems are already building critical infrastructure in the harshest outdoor environments, starting with large-scale solar. The founding team https://www.gritt.ai/team includes experts in robotics and AI from Carnegie Mellon, Stanford, and MIT. Gritt is backed by Obvious Ventures, Union Square Ventures, First Round Capital, Climactic, Congruent Ventures, and other leading firms.
Role: Software - ML Infrastructure Engineer
Location: SF Bay Area (in-person)
About the role
We’re looking for an experienced ML Infrastructure Engineer to join our team. As an early member, you will play a pivotal role in architecting scalable cloud infrastructure for our AI and data pipelines. You'll need to thrive in a fast-paced startup environment where you'll wear multiple hats and have a direct impact on our product's evolution. Ideally, you have a proven track record of developing and deploying high-performance ML and cloud pipelines in production, and you're passionate about pushing the boundaries of what's possible in robotics with AI.
What you’ll get to work on
- Develop and deploy scalable AI training and validation pipelines in the cloud.
- Spin up distributed pipelines for data ingestion, pre-processing, training and evaluation.
- Deploy monitoring and CI/CD pipelines.
- Enable large-scale evaluation of AI models via cloud-based metrics.
- Enable large-scale evaluation of autonomy software and models via simulations in the cloud.
- Optimize performance, I/O and GPU utilization.
- Build tooling and dashboards for rapid experimentation, orchestration and visualization.
- Work with other teams to integrate cloud tooling into workflows.
What we look for
- Degree in computer science or related engineering disciplines (or equivalent experience).
- 4+ years of experience deploying high-performance ML pipelines in production.
- Proficient in Python and comfortable with C++/Go.
- Experience with ML frameworks like PyTorch.
- Experience with IO and data-loading workflows, including formats like Parquet, HDF5, TFRecord etc.
- Experience with deploying on cloud platforms like AWS, GCP or Azure.
- Experience with tooling like Docker, Kubernetes, and Airflow.
- Should be comfortable taking ownership of tasks with light supervision.
- Must have excellent problem-solving skills.
- Legally authorized to work in the United States.