Senior Software Engineer, AI Infra

We are looking for a Senior Software Engineer to build scalable AI infrastructure, supporting the training, validation, and production workflows for robotics. You will help establish robust systems for next-generation AI-driven robots.

Responsibilities:

• Develop Python-based training infrastructure for stable and accelerated model training.
• Build tools for data collection, training, and deployment of neural networks on RoboForce robots.
• Optimize deployment pipelines for hardware platforms and maintain distributed systems.

Qualifications:

• Bachelor’s or Master’s degree in Computer Science or related field with 8+ years of experience.
• Proficiency in C++, Python, and ML frameworks (e.g., PyTorch, JAX).
• Experience managing cloud infrastructure (AWS, Azure, GCP) and data stores (e.g., Postgres, MySQL, ElasticSearch, Redis).
• Must be on-site with no remote work option.

Preferred Skills:

• Expertise in profiling and optimizing CPU-GPU interactions.
• Experience scaling neural network training jobs and GPU programming with CUDA.
• Proven ability to develop annotation and dataset management tools.