Senior Staff Machine Learning Engineer

You will design and implement machine-learning solutions across robotic perception, decision-making, action scoring, and prediction. You will own data curation, labeling, training, evaluation, deployment, and iteration; build scalable PyTorch pipelines; optimize reliability and performance; and establish model, dataset, and ML operations practices.

Responsibilities

• Design and implement machine-learning solutions across perception, decision-making, action scoring, and predictive modeling
• Own data curation, labeling, training, evaluation, deployment, and iteration
• Build scalable training and inference pipelines with PyTorch
• Integrate machine learning into real-time robotic systems
• Use profiling, monitoring, and experiments to optimize performance and reliability
• Ensure reproducibility, traceability, and modularity across training and serving pipelines
• Maintain production-quality Python and C++ code
• Establish practices for model versioning, dataset management, and ML operations
• Curate datasets, develop machine-learning models, and build data flywheels

Requirements

• Degree in Computer Science, Electrical Engineering, or Mathematics
• 5+ years of industry experience applying machine learning to production systems
• Strong Python and PyTorch experience
• Experience with classification, regression, ranking, segmentation, and structured prediction
• Experience profiling, debugging, and optimizing models and pipelines
• Experience designing reliable systems from model training through field deployment
• Experience with AWS, GCP, Azure, or cloud infrastructure
• Experience with Docker
• Familiarity with Linux, Git, CI/CD, testing, and code review