Drimva

Skild AI

Software Engineer, AI Training and Infrastructure

$100K–$300K

San Mateo, CA +3 · Full-time · Posted Sep 23, 2024

Location
San Mateo, CA
Bengaluru, Karnataka, India
Pittsburgh
San Francisco
Experience
Minimum of 3 years of industry experience.
Education
BS, MS or higher degree in Computer Science, Robotics, Engineering or a related field, or equivalent practical experience.

About the job

Build and optimize scalable software infrastructure and training pipelines for large-scale AI models used in real-world robotics.

What you'll do

  • Develop and maintain distributed training pipelines and frameworks, including data preprocessing, orchestration, and model evaluation.
  • Improve training performance, resource utilization, scalability, and reliability by monitoring systems and addressing bottlenecks.
  • Collaborate with researchers and machine learning engineers to integrate algorithms and maintain robust infrastructure, automated testing, and continuous integration.

What they ask for

  • Proficiency in Python, C++, or similar, and at least one deep learning library.
  • Strong background in distributed computing, parallel processing, large-scale datasets, and data preprocessing.
  • Deep understanding of machine learning techniques and models.
  • Experience with machine learning infrastructure, software engineering principles, and continuous integration and automated testing.

Nice to have

  • Experience with cloud-based training environments such as AWS, Google Cloud, or Azure.

Pay

$100,000 — $300,000 USD (base salary)

Tools

Python, C++, PyTorch, TensorFlow, JAX, AWS, Google Cloud, Azure

AI focus

Develop infrastructure and pipelines for training large-scale AI models for robotics, including data preparation, model evaluation, and deployment.

Model-written summary · the original posting is authoritative

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