Senior Software Engineer, AI Training & Infrastructure

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.
• Minimum Requirements
• 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. Experience can be concurrent.
• Must also have two (2) years of experience designing and operating distributed training pipelines at scale, including data preprocessing, orchestration, and evaluation. Experience can be concurrent.
• Must have any experience with each of the following: (i) Python or C++ and at least one deep learning library (e.g., PyTorch, TensorFlow, or JAX); and (ii) CI/CD and automated testing for ML/infra services. Experience can be concurrent.
• Apply online at skild.ai/career.
• Base Salary Range
• $200,000 - $300,000 USD

Additional Information

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 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 (e.g., ffmpeg). Specific duties include: (i) architecting, building, and maintaining distributed training pipelines and frameworks spanning data ingest/preprocessing, large-scale training, and evaluation; (ii) optimizing training performance and resource utilization by identifying bottlenecks and implementing improvements in data loading, I/O, caching, sharding, and prefetching; (iii) integrating state-of-the-art ML techniques into production training systems in collaboration with research/ML teams; (iv) implementing monitoring, logging, alerting, automated testing, and CI/CD for reliable training operations; and (v) developing developer tooling and documentation, including dashboards and utilities, to streamline experimentation at scale and improve engineer productivity.

Your Match

How well this role fits your profile.

Company Intel

What employees say

Worked at skildai-careers? Share your experience