Software Engineer, AI Inference

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 optimize runtime AI inference pipelines for real-world robotic deployment.
• Build infrastructure, frameworks, and tooling to enable reliable integration of models into robotic systems.
• Formulate specialized optimization solutions for various inference paradigms and scenarios.
• Adapt optimization solutions to various compute, hardware, and networking constraints.

Requirements

• BS, MS or higher degree in Computer Science, Robotics, Engineering or a related field
• Minimum of 3 years of industry experience
• Proficiency in C, C++, Rust, Go, or Python
• Experience with deep learning libraries such as PyTorch, TensorFlow, or JAX
• Deep understanding of low-level systems concepts like multithreading, networking, and memory management

Preferred Qualifications

• Experience with CUDA
• Deep understanding of state-of-the-art machine learning techniques and models
• Experience optimizing various machine learning architectures
• Experience with machine learning compilers
• Experience optimizing model inference for robotic systems deployment