Machine Learning Engineer - Vision
Apply
Apply for this job directly on SHORTList.
Referral
Share your custom referral link for this job with qualified candidates. Earn the referral you lead to a hire.
As a Senior Machine Learning Engineer, you will own the ML lifecycle for the detection and segmentation models at the core of our manufacturing intelligence pipeline. You will research, develop, and deploy cutting-edge object detection and segmentation models for layout analysis, document understanding, and semantic part understanding. You will work alongside the core engineering team to build and maintain annotation tooling, implement active learning loops, and engineer synthetic data augmentation strategies. Additionally, you will develop evaluation frameworks to capture metrics that extend beyond mAP/IoU, precisely quantifying system behavior and user impact. Collaborating with other members of the machine learning team, you will set the technical and product roadmaps for the AI platform. You will also focus on improving accuracy, as every percentage point of accuracy maps to significant time savings at our scale.
What We're Looking For
- 5-8 years of professional experience building and shipping computer vision models, with an emphasis on detection and/or segmentation.
- Deep expertise in modern architectures using transformer backbones (ViT, Swin, etc.).
- Strong Python and PyTorch fluency: You've written custom training loops, loss functions, and data loaders from scratch when needed.
- Production deployment ownership: You've shipped models to production and have been responsible for endpoint and model health.
- MS or PhD in Computer Science, Electrical Engineering, or related field preferred; equivalent industry experience valued equally.
Bonus Points
- You have a passion for manufacturing and believe that the industry needs better software.
- Previously worked in aerospace, defense, or manufacturing, and have experience working with manufacturing data.
- Published research, achieved SOA results on relevant benchmarks, or contribute to open-source frameworks.
- Prior experience working in a high-ownership startup environment.
Apply
Apply for this job directly on SHORTList.
Referral
Share your custom referral link for this job with qualified candidates. Earn the referral you lead to a hire.