Senior/Staff Machine Learning Engineer

Job Overview

Senior/Staff Machine Learning Engineer at Dexterity, where you will work on challenges related to robot task and action planning. You will leverage techniques from machine learning to solve hard sequential decision problems that require reasoning about the physical world and its dynamics. This is a remote work engineering position focused on developing Dexterity’s technology foundations in Physical AI.

About Dexterity

At Dexterity, we believe robots can positively transform the world. Our breakthrough technology frees people to do the creative, inspiring, problem-solving jobs that humans do best by enabling robots to handle repetitive and physically difficult work. We’re starting with warehouse automation, where the need for smarter, more resilient supply chains impacts millions of lives and businesses worldwide. Dexterity's full-stack robotics systems pick, move, pack, and collaborate with human-like skill, awareness, and learning capabilities. Our systems are software-driven and hardware-agnostic and have already picked 100+ million goods in production. We’re customer-obsessed and one of the fastest-growing companies in robotics, backed by world-class investors such as Kleiner Perkins, Lightspeed Venture Partners, and Obvious Ventures.

Key Responsibilities

• Design and implement machine learning solutions across Dexterity’s robotics stack, including perception, decision-making, action scoring, and predictive modeling
• Own the full ML development cycle: data curation, labeling, training, evaluation, deployment, and iteration
• Build performant training and inference pipelines using PyTorch, with production-readiness and scalability in mind
• Collaborate closely with robotics, data platform, and simulation teams to integrate ML into real-time, latency-sensitive robotic systems
• Use profiling, monitoring, and experiments to optimize model performance and reliability
• Ensure reproducibility, traceability, and modularity across training and serving pipelines
• Maintain clean, production-quality code in Python (and C++ where required)
• Help establish best practices for model versioning, dataset management, and ML operations

Required Qualifications

• Degree in Computer Science, Electrical Engineering, or Mathematics
• 5+ years of industry experience applying machine learning to real-world, production systems
• Strong Python skills and deep experience with PyTorch
• Ability to work fluently across ML tasks, e.g., classification, regression, ranking, segmentation, and structured prediction
• Strong engineering background with experience profiling, debugging, and optimizing model and pipeline performance
• Proven ability to design and maintain reliable systems, from model training to field deployment
• Experience with cloud-based infrastructure (AWS, GCP, Azure) and containerized environments (Docker)
• Familiarity with Linux, Git, CI/CD) and software development best practices (unit/acceptance/integration testing, code reviews)

Nice to Have

• Prior experience in robotics, autonomous systems, or real-time ML applications
• Exposure to multimodal data (e.g., RGBD, force-torque, pose estimates, telemetry)
• Experience deploying models using serving stacks like NVIDIA Triton, TorchServe, or custom low-latency frameworks
• Background in computer vision, geometry