Software Engineer, AI Training and Infrastructure
Core
Building scalable software infrastructure and training pipelines for general-purpose robotic intelligence models.
Role type
Senior IC software engineer (AI training infrastructure)
Builds
Distributed training pipelines, frameworks, and tooling for large-scale AI models
Domain
Robotics + Machine Learning Infrastructure
Deliverable
production ML models
Required skills
Python, C++, distributed computing, parallel processing, large-scale data handling, software engineering principles, system design, automated testing
Preferred skills
PyTorch, TensorFlow, JAX, cloud platforms (AWS/GCP/Azure), CI/CD frameworks
Technologies
Python, C++, PyTorch, TensorFlow, JAX, AWS, Google Cloud, Azure
Responsibilities
Develop and maintain robust, scalable, and distributed training pipelines; Optimize training processes for performance and resource utilization; Collaborate with researchers to integrate state-of-the-art algorithms; Monitor and analyze training to identify bottlenecks; Ensure robustness and reliability of training infrastructure
Seniority
Senior, hands-on IC
