Remote job
Senior Software Engineer - Scalable Machine Learning
Job details
About this role
Role overview Build the software and cloud infrastructure that helps machine learning teams train models efficiently on large volumes of multi-sensor data. The role spans data ingestion, workflow orchestration, distributed training, and MLOps, with close collaboration across perception and robotics teams to deliver production-ready systems.
Responsibilities - Design and operate cloud services for data ingestion, workflow orchestration, and distributed model training. - Develop systems to generate, manage, and process large multi-sensor training datasets. - Improve MLOps workflows for model artifacts, experiment tracking, deployment, and automated testing. - Integrate machine learning pipelines with cross-functional engineering and robotics teams. - Mentor junior engineers and contribute technical guidance to improve software development practices.
Requirements - Relevant degree and experience: a bachelor’s degree plus 4 or more years of related work, a master’s degree plus 2 or more years, or a PhD. - Record of delivering complex software projects from initial design through production. - Experience building machine learning data pipelines, data generation systems, and distributed training workflows. - Experience designing and operating large-scale distributed systems and cloud infrastructure. - Strong Python and/or C++ skills for production-quality software.
Nice to have - Graduate study in computer science, electrical engineering, robotics, or a related discipline. - Experience with distributed ML tools such as Ray, PyTorch, Dagster, LakeFS, or PyArrow. - Familiarity with multi-sensor data processing, sensor fusion, computer vision, or autonomous systems. - Knowledge of safety-critical software practices in automotive or robotics settings.
Benefits and work setup The listing describes medical, dental, and vision coverage; employer-matched retirement savings; paid parental and medical leave; vacation and 15 paid holidays; wellness and commuter support; professional development reimbursement; and family care programs. It gives a California base salary range of $179,200–$268,800, with compensation based on role-related factors. Visa sponsorship is available; permanent US work authorization is required.