July, 2026
Machine Learning Engineer
Pittsburgh, PA | Full-Time | On-site
About Us
Shelfmark is the quality control platform that lets manufacturers catch defects the moment they happen. Our systems pair high-resolution line scan cameras with embedded software and machine learning to inspect products as they move down the line - flagging flaws in real time, at production speed, so our customers ship quality with confidence instead of relying on slow, manual spot checks.
As a Machine Learning Engineer, you'll own the models at the heart of that inspection pipeline. You'll work hands-on with imagery captured directly off the line, building and tuning the computer vision and anomaly-detection models that decide what's good and what isn't - then getting them running fast and reliably on the edge hardware that sits next to the camera. You'll work closely with our hardware and embedded engineers to ensure our models hold up against real world conditions.
What You'll Do
Build, train, and evaluate ML models for defect detection and classification on imagery captured from live inspection lines.
Develop ML models for anomaly detection and classification in cases where labeled defects are rare or hard to define.
Optimize and deploy models to run on edge hardware at the inspection line, balancing accuracy against latency and throughput constraints.
Partner with hardware and embedded systems engineers to integrate models into the end-to-end inspection pipeline, from camera capture to real-time decision.
Establish and improve the data workflow - labeling, dataset curation, augmentation, and retraining loops - to keep models sharp as products and conditions change.
Diagnose model performance issues in production and on-site, using real line data to drive improvements.
Help define ML standards, tooling, and best practices that the broader team will build on.
What We’re Looking For
Experience training, evaluating, and deploying deep learning models using Pytorch or Tensorflow, preferably for computer vision applications.
Hands-on experience with classical image processing, unsupervised/self-supervised learning methods, preferably applied to anomaly detection. Solid grounding in classical computer vision and statistical modeling.
Experience with or strong interest in vision-language models for tasks like zero/few-shot classification, and prompt driven anomaly detection.
Familiarity with model optimization for edge deployments – quantization, ONNX/TensorRT, and profiling models under latency and memory constraints.
A pragmatic, results-oriented mindset - comfortable working with messy real-world data and iterating quickly.
Willingness to work on-site and collaborate closely with hardware and embedded teammates.
Bonus: experience in manufacturing, industrial inspection, or other real-time/high-throughput vision systems.
Compensation: Machine Learning Engineer will receive a competitive salary and healthcare benefits package and would be considered for early employee equity compensation.
Location: The ideal candidate would be located in Pittsburgh, PA and willing to co-locate in person in the company’s Uptown office.
Skills Required
- Experience training, evaluating, and deploying deep learning models using PyTorch or TensorFlow
- Hands-on experience with classical image processing and unsupervised/self-supervised learning methods for anomaly detection
- Familiarity with model optimization for edge deployments (quantization, ONNX, TensorRT) and profiling models under latency and memory constraints
- Experience optimizing and deploying models to run on edge hardware and integrating with embedded systems
- Ability to diagnose model performance issues in production using real line data
- Willingness to work on-site in Pittsburgh and collaborate closely with hardware and embedded teammates
- Experience with or strong interest in vision-language models for zero/few-shot classification and prompt-driven anomaly detection
- Experience in manufacturing, industrial inspection, or real-time/high-throughput vision systems (bonus)
What We Do
Shelfmark is a production intelligence platform for continuous-flow manufacturing. Its managed AI systems automate in-line visual inspection, identify defects, predict recurring problems, and explain their causes. By combining hardware, software, and ongoing support, Shelfmark helps manufacturers improve product quality, reduce manual inspection labor, minimize waste and rework, and increase production efficiency across industrial operations.
