Demo

Senior Software Engineer, AI Training & Infrastructure

Skild AI
San Mateo, CA Full Time
POSTED ON 4/14/2026
AVAILABLE BEFORE 5/8/2026
Company Overview

At Skild AI, we are building the world's first general purpose robotic intelligence that is robust and adapts to unseen scenarios without failing. We believe massive scale through data-driven machine learning is the key to unlocking these capabilities for the widespread deployment of robots within society. Our team consists of individuals with varying levels of experience and backgrounds, from new graduates to domain experts. Relevant industry experience is important, but ultimately less so than your demonstrated abilities and attitude. We are looking for passionate individuals who are eager to explore uncharted waters and contribute to our innovative projects.

Position Overview

Skild AI, Inc. seeks a Senior Software Engineer, AI Training & Infrastructure in San Mateo, CA. You will be responsible for building and scaling training infrastructure and tools that support the full ML lifecycle—data preparation, training orchestration, evaluation, and deployment—for real-world robotics applications. This includes performance, reliability, observability, and developer productivity across distributed training systems, as well as data processing for multimodal datasets, performance tuning of training jobs, and media processing/compression.

Responsibilities

  • Architecting, building, and maintaining distributed training pipelines and frameworks spanning data ingest/preprocessing, large-scale training, and evaluation.
  • Optimizing training performance and resource utilization by identifying bottlenecks and implementing improvements in data loading, I/O, caching, sharding, and prefetching.
  • Integrating state-of-the-art ML techniques into production training systems in collaboration with research/ML teams.
  • Implementing monitoring, logging, alerting, automated testing, and CI/CD for reliable training operations.
  • Developing developer tooling and documentation, including dashboards and utilities, to streamline experimentation at scale and improve engineer productivity.

Preferred Qualifications

  • Must have a master's degree (or foreign equivalent) in Computer Science, Robotics, Engineering, or a related field and two (2) years of experience in machine learning infrastructure.
  • Must also have two (2) years of experience designing and operating distributed training pipelines at scale, including data preprocessing, orchestration, and evaluation.
  • Must have any experience with each of the following: Python or C and at least one deep learning library (e.g., PyTorch, TensorFlow, or JAX); and CI/CD and automated testing for ML/infra services.
  • Must have knowledge of: Optimizing data loading and I/O for deep learning workloads (e.g., PyTorch DataLoader, sharding, prefetching, or caching); processing multimodal datasets and formats (e.g., HDF5, TFRecord, Parquet, or equivalent) and image processing/compression (e.g., OpenCV or ffmpeg); cloud-based training in AWS, Google Cloud, or Azure; Implementing monitoring, logging, and alerting for training systems; Linux OS fundamentals and operation at large scale; distributed systems and ML training techniques/models; and core software engineering principles, including algorithms, data structures, and system design.

Base Salary Range

$200,000—$300,000 USD

Salary : $200,000 - $300,000

If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

What is the career path for a Senior Software Engineer, AI Training & Infrastructure?

Sign up to receive alerts about other jobs on the Senior Software Engineer, AI Training & Infrastructure career path by checking the boxes next to the positions that interest you.
Income Estimation: 
$119,030 - $151,900
Income Estimation: 
$149,493 - $192,976
Income Estimation: 
$101,387 - $124,118
Income Estimation: 
$119,030 - $151,900
Employees: Get a Salary Increase
View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

Job openings at Skild AI

  • Skild AI San Francisco, CA
  • Company Overview At Skild AI, we are building the world's first general purpose robotic intelligence that is robust and adapts to unseen scenarios without ... more
  • 8 Days Ago

  • Skild AI San Francisco, CA
  • Company Overview At Skild AI, we are building the world's first general purpose robotic intelligence that is robust and adapts to unseen scenarios without ... more
  • 9 Days Ago

  • Skild AI San Mateo, CA
  • Company Overview At Skild AI, we are building the world's first general purpose robotic intelligence that is robust and adapts to unseen scenarios without ... more
  • 9 Days Ago

  • Skild AI Pittsburgh, PA
  • Company Overview At Skild AI, we are building the world's first general purpose robotic intelligence that is robust and adapts to unseen scenarios without ... more
  • 10 Days Ago


Not the job you're looking for? Here are some other Senior Software Engineer, AI Training & Infrastructure jobs in the San Mateo, CA area that may be a better fit.

  • Skild AI San Francisco, CA
  • Company Overview At Skild AI, we are building the world's first general purpose robotic intelligence that is robust and adapts to unseen scenarios without ... more
  • 12 Days Ago

  • Fireworks AI San Mateo, CA
  • About Us: At Fireworks, we’re building the future of generative AI infrastructure. Our platform delivers the highest-quality models with the fastest and mo... more
  • 12 Days Ago

AI Assistant is available now!

Feel free to start your new journey!