Benchmark Construction Technology
Posted 4mo ago

Lead ML Engineer - Computer Vision

Benchmark Construction Technology
India
RemoteFull Time
Responsibilities
  • Architect ML
  • Mentor engineers
  • Ensure reliability
Requirements
  • 8+ years ML engineering with computer vision
  • Expert Python with PyTorch
  • Experience in MLops
  • OCR/document pipelines
  • Linux and Docker
  • Scalable ML systems
Technical tools mentioned
PythonPyTorchFastAPIOpenCVDockerMLflowClearML

Job description

About the role

You’ll own and expand our AI-enabled plan ingestion pipeline—from raw PDFs to clean, dependable outputs that power takeoff, design, and downstream automation.

What you'll do

  • Architect, build, and deploy ML systems for object detection, segmentation, document understanding, and information extraction from complex building plans
  • Drive R&D across computer vision and/or NLP as applied to construction documents and related data
  • Develop multi-modal approaches that combine vision with language features for document analysis (LLMs/NLP). Familiarity with different RAG methods is a plus
  • Mentor junior engineers and help raise the bar on execution, code quality, and decision-making
  • Ensure systems are reliable, maintainable, testable, and observable in production

Qualifications

  • 8+ years of experience in Machine Learning Engineering, with deep focus in computer vision
  • Expert Python skills, with strong experience in PyTorch and FastAPI
  • Proven work with segmentation models, detection frameworks, and OCR/document pipelines
  • Strong image processing fundamentals (OpenCV, geometric transforms, preprocessing, etc.)
  • Comfortable in Linux; hands-on with Docker/containerization for deployment
  • Experience building and maintaining MLops infrastructure (e.g., ClearML, MLflow)
  • Ability to design scalable, high-performance ML systems (not just prototypes)
  • Strong problem-solving ability and comfort operating independently in a fast-paced environment

What Makes You a Great Fit

  • High agency: you ship, iterate, and fix what’s broken—without waiting for permission
  • You’ve delivered ML systems end-to-end in production (data → model → deployment → iteration)
  • You’re a strong engineer who cares about getting better—trajectory matters here
  • You communicate clearly, especially with a remote and cross-cultural team
  • You’re gritty and genuinely enjoy solving hard problems

Bonus Points

  • Experience building systems for document reading / understanding
  • Ambitious, driven, and opinionated about quality and craft
  • Experience with deep learning inference frameworks (e.g., TensorRT, PyTorch compile) and optimizing inference performance
  • You keep up with current ML/AI developments and can apply them pragmatically

What We Offer

  • Competitive salary + meaningful equity
  • Comprehensive benefits (health, dental, vision)
  • Professional development budget (courses, conferences, research exploration)
  • Real-world ML problems with direct impact
  • High growth potential

About Benchmark Construction Technology

AI-powered platform for construction estimating and design.

Similar jobs

Lead ML Engineer roles
1y
Save
Mark Applied
Hide
ML Lead Engineer - GenAI Project - Offshore
Bangalore, Karnataka, India
OnsiteFull Time
Ascendion
Ascendion: AI-native digital engineering and software development services provider.
6+ YOE6+ years of experience building and deploying ML systems; strong ML frameworks; Python; MLops; cloud and containers; generative AI.
Python, PyTorch, TensorFlow, Scikit-learn, Pandas, NumPy, Azure Cloud, Generative AI Frameworks, FAST API, Flask
9mo
Save
Mark Applied
Hide
Lead AI ML Engineer
Noida, India
HybridFull Time
3Pillar Global
3Pillar Global: Develops custom software products and digital solutions for companies.
8+ YOE8+ years IT with 4+ years in AI/ML; strong Python and ML frameworks; NLP libraries; AI/ML algorithms; MLOps and CI/CD for AI; leadership and mentoring.
Python, TensorFlow, PyTorch, Scikit-Learn, spaCy, Hugging Face Transformers, OpenCV, CI/CD
3mo
Save
Mark Applied
Hide
Lead AI/ML Software Use Cases Validation Engineer
Bangalore, Karnataka, India
₹3246k-₹4637k/yr OnsiteFull Time
AMD
AMDNASDAQ: AMD: Designs and manufactures computer processors and graphics technology.
8+ YOE8–12 years in AI/ML software validation or performance engineering; strong Python scripting and test automation; deep learning and LLMs; ROCm validation; Linux/Ubuntu/Yocto experience; end-to-end AI pipeline validation; benchmarking and optimization; strong collaboration and leadership skills.
Python, PyTorch, TensorFlow, ONNX, ROCm, HIP, CUDA, OpenCL, Ubuntu, Yocto, Linux