RoboForce
Senior Software Engineer, AI InfraMilpitas, CA
RoboForce, Milpitas, California, United States, 95035
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). Requires 5 days/week in-office collaboration with the teams. 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.
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). Requires 5 days/week in-office collaboration with the teams. 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.