Drimva

Gritt Robotics Inc

ML Infrastructure Engineer

South San Francisco · On-site · Full-time · Posted May 26, 2026

Experience
4+ years deploying high-performance ML pipelines in production.
Education
Degree in computer science or related engineering disciplines (or equivalent experience).
Visa sponsorship
Legally authorized to work in the United States.

About the job

Builds and operates cloud infrastructure and pipelines for training, evaluating, and monitoring AI models and autonomy software for robotics.

What you'll do

  • Develop scalable cloud pipelines for data ingestion, preprocessing, AI training, validation, and evaluation.
  • Deploy monitoring and CI/CD pipelines, and optimize performance, I/O, and GPU utilization.
  • Build experimentation, orchestration, and visualization tools, and integrate cloud tooling with other teams.

What they ask for

  • Degree in computer science or a related engineering discipline, or equivalent experience.
  • 4+ years of experience deploying high-performance ML pipelines in production.
  • Proficient in Python and comfortable with C++/Go; experience with PyTorch.
  • Experience with data-loading workflows and formats, cloud platforms, and Docker, Kubernetes, and Airflow.

Tools

Python, C++, Go, PyTorch, Parquet, HDF5, TFRecord, AWS

AI focus

Develop and operate AI training, validation, and evaluation pipelines, including cloud-based evaluation of models and autonomy software through simulation.

Model-written summary · the original posting is authoritative

First seen Oct 29, 2026 · Last verified Nov 5, 2026 (UTC)