Machine Learning / Computer Vision Engineer

Responsibilities

• Build computer vision and visual representation learning pipelines for robotic manipulation, including RGB, RGB-D, depth, segmentation, pose, keypoint, and object-centric representations.
• Develop visual models that support reinforcement learning and imitation learning policies, including end-to-end visuomotor policies that map visual observations to robot actions.
• Improve our data pipeline for vision-based manipulation policies through domain randomization, photorealistic rendering, synthetic data generation, sensor noise modeling, and real-world fine-tuning.
• Design and train perception models that are robust to lighting changes, camera viewpoint shifts, texture variation, clutter, occlusion, object instance variation, and imperfect calibration.
• Evaluate learned visual representations and policies on real robotic manipulation tasks, identify failure modes, and iterate on models, data, and training procedures.
• Collaborate with robotics, robot learning, and simulation engineers to define the perception strategy for robotic manipulation.
• Set up, calibrate, and evaluate camera and depth sensing systems when needed, with an emphasis on how sensor choices affect learned policies and real-world robustness.

Qualifications

• Ph.D. in computer vision or 3+ years of experience working on a computer vision product.
• Strong background in machine learning for computer vision, especially deep learning-based visual perception.
• Experience training modern computer vision models in Jax, PyTorch or similar frameworks.
• Practical experience with visual representation learning, object detection, segmentation, pose estimation, depth estimation, tracking, or 3D perception.
• Strong Python programming skills.
• Ability to move fluidly between research code and production-quality systems.
• Strong understanding of how data distribution, sensor noise, calibration, lighting, and scene variation affect model performance.