ML & Cloud Infrastructure Engineer

Gritt Robotics; South San Francisco (CA); On-site; USD 140,000 - 180,000; Full time

Job summary

Gritt Robotics in the SF Bay Area is seeking an experienced ML & Cloud Infrastructure Engineer to architect scalable cloud infrastructure for AI and data pipelines. As an early team member, you will directly influence product evolution and deployment of high-performance ML pipelines in production.

You will own cloud tooling, CI/CD, data ingestion and model evaluation pipelines, with emphasis on portability, reliability and performance across GPU-backed workloads in a fast-growing robotics

Qualifications

• 4+ years 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 (Parquet, HDF5, TFRecord).
• Experience deploying on cloud platforms like AWS, GCP or Azure.
• Experience with tooling like Docker, Kubernetes, and Airflow.
• Legally authorized to work in the United States.
• Degree in computer science or related engineering disciplines.

Responsibilities

• Develop and deploy scalable ML pipelines in the cloud.
• Spin up distributed data ingestion, training and evaluation pipelines.
• 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.

Skills

ML pipelines

Python

C++/Go

PyTorch

Parquet/HDF5/TFRecord

Cloud platforms

Docker

Kubernetes

Airflow

Ownership

Problem-solving

US work authorization

Education

B.S. in Computer Science

Tools

Docker

Kubernetes

Airflow

Job description

Gritt 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 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 & Cloud Infrastructure

Location: SF Bay Area (in-person)

About the role

We’re looking for an experienced ML & Cloud 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.