Skild AI · 5 hours ago
Software Engineer, AI Training and Infrastructure
San Francisco Bay Area
Full-time
Onsite
Mid Level
$100K/yr - $300K/yr
3+ years expSkild AI is building the world's first general purpose robotic intelligence that adapts to unseen scenarios. They are seeking a Software Engineer to develop and optimize software infrastructure and tools for training AI models, focusing on building scalable training pipelines and collaborating with researchers and engineers.
Computer Software
Responsibilities
Develop and maintain robust, scalable, and distributed training pipelines (data preprocessing, training orchestration, and model evaluation) and frameworks for large-scale AI models
Optimize training processes for performance and resource utilization, ensuring scalability and reliability
Collaborate with researchers and machine learning engineers to integrate state-of-the-art algorithms and techniques into training pipelines
Monitor and analyze training, identifying bottlenecks and proposing solutions to improve efficiency and performance
Ensure the robustness and reliability of the training infrastructure, including automated testing and continuous integration
Qualification
Required
Develop and maintain robust, scalable, and distributed training pipelines (data preprocessing, training orchestration, and model evaluation) and frameworks for large-scale AI models
Optimize training processes for performance and resource utilization, ensuring scalability and reliability
Collaborate with researchers and machine learning engineers to integrate state-of-the-art algorithms and techniques into training pipelines
Monitor and analyze training, identifying bottlenecks and proposing solutions to improve efficiency and performance
Ensure the robustness and reliability of the training infrastructure, including automated testing and continuous integration
Preferred
BS, MS or higher degree in Computer Science, Robotics, Engineering or a related field, or equivalent practical experience
Minimum of 3 years of industry experience
Proficiency in Python, C++, or similar and at least one deep learning library such as PyTorch, TensorFlow, JAX, etc
Strong background in distributed computing, parallel processing techniques, handling large-scale datasets and data preprocessing
Deep understanding of state-of-the-art machine learning techniques and models
Experience with cloud-based training environments (AWS, Google Cloud, Azure)
Experience in developing and maintaining software tooling and infrastructure for machine learning
Deep understanding and practical experience with software engineering principles, including algorithms, data structures, and system design
Experience with continuous integration and automated testing frameworks
Company
Skild AI
Building general purpose robotic intelligence.
11-50 employees
H1B Sponsorship
Skild AI has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (14)
2024 (2)
Funding
Current Stage
Early StageRecent News
2025-10-03
2025-09-27
2025-09-26
Company data provided by crunchbase