Applied ML Engineer: Production CV for Construction

SketchDeck.AI Inc.; Atlanta (GA); Hybrid; USD 120,000 - 160,000; Full time

Job summary

SketchDeck.ai seeks an experienced Machine Learning / AI Engineer to design, train, evaluate, deploy, and improve computer vision systems at the core of our products. You will own the end-to-end ML lifecycle from dataset development to production feedback.

Responsibilities span dataset creation, experimentation, model architecture, training, inference, and deployment, with focus on robust performance across diverse construction drawings and customer environments.

Qualifications

• 4+ years of professional experience in applied machine learning or computer vision.
• Strong production experience building CV systems for detection, segmentation, or classification.
• Strong PyTorch experience including training, evaluation, and optimization.

Responsibilities

• Design, train, evaluate, and improve computer vision models for object detection, segmentation, and document understanding.
• Develop production models using PyTorch and YOLO-family architectures where appropriate.
• Improve inference performance, monitoring, and cost across production workloads.

Skills

Python

PyTorch

Computer Vision

Data Analysis

Experimentation

Team Collaboration

Communication

Education

Master’s or PhD in CS/ML or related

Tools

Docker

MongoDB

AWS/GCP

Git

OpenCV

Job description

SketchDeck.ai seeks an experienced Machine Learning / AI Engineer to design, train, evaluate, deploy, and improve computer vision systems at the core of our products. You will own the end-to-end ML lifecycle from dataset development to production feedback.

Responsibilities span dataset creation, experimentation, model architecture, training, inference, and deployment, with focus on robust performance across diverse construction drawings and customer environments.