Software Engineer, AI Training and Infrastructure

About the Company

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.

Responsibilities

• Develop and maintain robust, scalable, and distributed training pipelines and frameworks for large-scale AI models
• Optimize training processes for performance and resource utilization
• Collaborate with researchers and machine learning engineers to integrate state-of-the-art algorithms
• Monitor and analyze training, identifying bottlenecks and proposing solutions
• Ensure the robustness and reliability of the training infrastructure, including automated testing and continuous integration

Requirements

• BS, MS or higher degree in Computer Science, Robotics, Engineering or a related field
• Minimum of 3 years of industry experience
• Proficiency in Python, C++, or similar
• Experience with at least one deep learning library such as PyTorch, TensorFlow, or JAX
• Strong background in distributed computing and parallel processing
• Experience with cloud-based training environments (AWS, Google Cloud, Azure)
• Deep understanding of software engineering principles, algorithms, and system design