Staff Machine Learning Software Engineer

Job description

Job responsibilities

• Design, build, and maintain scalable ML pipelines for data ingestion, model training, evaluation, and related research workflows.
• Build software systems that allow ML workloads to scale to larger datasets, larger models, and more experiments.
• Develop clean and reusable interfaces between data, models, training, evaluation, and downstream robotics workflows.
• Own and improve the health of the MEGA codebase, including software architecture, testing, reliability, maintainability, and engineering standards.
• Identify and resolve performance, reliability, and usability bottlenecks across ML workflows.
• Build and maintain infrastructure that supports multiple researchers and ML projects without unnecessarily slowing down iteration.
• Work closely with researchers to understand the requirements of new models and experiments, and translate those requirements into practical software solutions.
• Build and use distributed training and data-processing pipelines for large models and large multimodal datasets.