ML Engineer (Computer Vision)

About AI71:

AI71 is an industry leader in artificial intelligence, delivering innovative solutions that empower developers, businesses and governments to solve complex challenges. AI71 builds secure, enterprise-ready applications powered by cutting-edge technology—tailored for knowledge workers and sector-specific needs. AI71 bridges the gap between advanced AI and real-world impact. Guided by a strong commitment to research and responsibility, we create transformative solutions that drive progress and empower communities.

The Role

We are seeking AI/ML Engineer with expertise in computer vision and image processing. The ideal candidate will develop and optimize deep learning models / VLMs for object detection, segmentation, classification, text extraction, and fine-tuning for domain-specific applications.

What You'll Do:

Develop & Deploy AI Models:

• Train and optimize object detection, segmentation, and classification models for architectural drawings.
• Fine-tune YOLO, R-CNN, and Transformer-based models to improve accuracy and efficiency for domain-specific tasks.
• Implement OCR and symbol recognition for automated compliance checking.
  
  

Image Processing & Enhancement:

• Develop lossless image compression and feature extraction techniques.
• Improve image pre-processing pipelines for architectural data quality.
• Utilize OpenCV and Deep Learning for drawing analysis.
  
  

Fine-Tuning & Model Adaptation:

• Customize and fine-tune pre-trained vision-language models (VLMs) while enabling their multimodal AI capabilities.
• Optimize model parameters, embeddings, and transfer learning strategies to enhance performance and efficiency.
• Continuously improve AI models based on real-world feedback, evolving datasets, and domain requirements.
  
  

AI Model Optimization & Deployment

• Optimize model architectures to reduce computational costs while maintaining accuracy.
• Deploy AI models on cloud or on-prem infrastructure with Docker/Kubernetes is a plus.
• Integrate AI workflows into AutoCAD and BIM environments is a plus.
  
  

What You'll Bring:

• Computer Vision & Deep Learning: OpenCV, YOLO, R-CNN, Mask R-CNN, EfficientNet
• OCR & Symbol Extraction: Tesseract, EasyOCR, NLP for document parsing
• Fine-Tuning & Transfer Learning: Vision-Language Models (VLMs), CLIP, Transformer-based architectures
• Programming: Python, C++
• Cloud & Deployment: AWS, Azure, GCP, TensorFlow Serving, ONNX
• Image Processing: Feature extraction, noise reduction, compression techniques
• 7+ years of Experience in the computer vision solutions
• Strong problem-solving skills in computer vision applications.
• Ability to optimize AI pipelines for real-world datasets.
• Experience integrating AI models into software applications.
• Familiarity with CAD tools (AutoCAD, Revit, DXF/DWF processing) is a plus.
  
  

Why AI71:

• Mission-Driven Work: Work on cutting-edge AI applications with a talented and passionate team, solving real-world challenges in critical sectors.
• Unparalleled Opportunity: This is a chance to innovate and solve real-world challenges using AI at a company with unique access to world-leading models and resources.
• Career Growth: We offer competitive compensation, benefits, and significant career growth opportunities as a foundational member of the team.
• World-Class Environment: Enjoy a flexible working environment and the latest tools & technologies needed to do your best work.