Lead Machine Learning Engineer (Computer Vision)

Hello everyone!

BUILDCHECK™—a dynamic, venture-backed Stanford University startup—is looking for a talented Machine Learning Engineer to drive innovation in the construction technology space. We’re on a bold mission: revolutionizing construction workflows by building the industry’s first AI-powered “spell-checker” for construction blueprints. Our solution helps architects, developers, and contractors detect costly errors before they become expensive problems.

Your responsibilities:

• Lead development of computer vision models for interpreting 2D construction drawings.
• Create optimized semantic segmentation, object detection, classification, and zero-shot learning pipelines specific to construction drawings analysis.
• Continuously evaluate and enhance model performance to ensure high accuracy and operational efficiency.
• Collaborate closely with Product and Engineering teams to integrate vision models into scalable backend systems.
• Stay current with the latest developments in computer vision and multimodal document analysis technologies.

Qualifications:

• Bachelor’s, Master’s, or PhD in Computer Science, Mathematics, Data Science, or a related field.
• 6+ years of hands-on experience in AI/ML, specifically with a strong focus on computer vision.
• Deep understanding of modern architectures of Neural Networks for CV problems.
• Proven expertise in semantic segmentation, object detection, and image classification (YOLO, SAM, SegFormer).
• Practical experience working with deep learning frameworks (PyTorch, TensorFlow, Hugging Face) and image processing libraries (OpenCV).
• Strong coding proficiency in Python.
• Experience deploying scalable ML pipelines using cloud infrastructure (AWS Lambda, AWS SageMaker).
• Excellent analytical skills and the ability to collaborate effectively across teams.

Preferred Skills:

• Experience with multimodal AI systems combining CV and NLP technologies.
• Familiarity with PDF layout analysis and document understanding tools.
• Knowledge of advanced inference optimization techniques (e.g., TensorRT, ONNX).
• Familiarity with model lifecycle management tools (MLFlow, Weights & Biases).

What else matters to us?

We’re looking for passionate individuals who thrive in startups, value impactful solutions, and enjoy ownership of projects that directly enhance our customers’ workflows.

Team and Culture:

Join our collaborative, rapidly growing team (currently ~15 people) distributed internationally, primarily in the US and Europe. You’ll work closely with the founders and an experienced engineering team that values agility, innovation, and practical problem-solving. Our work is guided by an exceptional advisory board comprising research directors and Nobel laureates from Stanford, MILA, ServiceNow, and DeepMind, providing strategic insight and domain expertise as we tackle complex challenges.

Interview process:

• Introduction interview
• Technical interview (system design & ML)
• Technical interview (coding & problem-solving)
• Founder cultural-fit interview

Interviews can be scheduled flexibly over one or several days. References will be requested and checked.

Compensation and Benefits:

• Competitive market salary
• Flexible remote work
• Opportunity to contribute to publications in peer-reviewed journals and company patents

Relocation to the U.S. (Bay Area) or Canada (Montreal) is possible, subject to Ukrainian martial law or other applicable legislation restricting international travel.

We’re excited to meet exceptional talent to help us redefine the construction industry’s future with AI!