Back to Jobs

Senior ML Engineer at Draftaid

Draftaid
Remote
Full-time
$145K–$177K
Estimated
Remote
Apply Now

Required Skills

Machine Learning
Self-Supervised Learning
PyTorch/TensorFlow
MLOps
Computational Geometry

Job Description

DraftAid.io is revolutionizing the mechanical engineering process with AI-powered CAD drawing automation. Our technology transforms 3D models into precise 2D fabrication drawings, helping engineers across automotive, construction, and manufacturing industries save time and reduce errors.

  • Think Gen-AI for blueprints. At DraftAid, we are building software that generates CAD manufacturing drawings.
  • Mechanical engineers and designers spend 40% of their time creating production drawings from their 3D models. It's a tedious process, error-prone, and often delays projects by weeks! DraftAid auto-generates these detailed drawings with a single click. Engineers designing anything from skyscrapers to car parts use DraftAid to speed up their workflow. With DraftAid, designers can model, and leave the drawings to AI!
  • We are hiring a senior ML engineer who’s excited to bring AI into the mechanical design space.
  • Develop and implement novel self-supervised learning approaches and prediction targets for 3D CAD data representation
  • Build end-to-end ML pipelines from data ingestion to production deployment
  • Work closely with software engineers to integrate ML models into our production systems
  • Continuously evaluate and optimize model performance against real-world engineering requirements
  • Stay current with the latest research and techniques in the field, adapting them to our unique challenges
  • 5+ years of experience in applied machine learning with a proven track record of shipping ML systems to production
  • Demonstrated expertise in developing embedding techniques and self-supervised learning approaches using transformers and diffusion models
  • Strong programming skills in Python and familiarity with ML frameworks such as PyTorch or TensorFlow
  • Experience with MLOps practices and tools for model deployment, monitoring, and management
  • Experience in fast-paced startup environments where pragmatic solutions take priority over academic perfection
  • Bachelor's degree in Computer Science, Machine Learning, or related field; advanced degree preferred
  • Experience applying ML techniques to 3D data, geometric shapes, or CAD models
  • Knowledge of computational geometry and its intersection with modern deep learning techniques
  • Experience with vector databases and similarity search infrastructure
  • Published research or contributions to the field of representation learning or geometric deep learning
  • Understanding of CAD systems, engineering workflows, or manufacturing processes
  • The opportunity to shape foundational technology and be a part of a groundbreaking startup from its early stages
  • A dynamic and supportive team environment where your contributions are valued and celebrated
  • Competitive salary and equity package
  • Flexible working hours
  • The chance to work on challenging projects that have the potential to transform manufacturing
  • The satisfaction of seeing your ML innovations directly impact and improve real-world engineering workflows

Job Details

Employment Type

Full-time

Salary Range

$145K–$177K

Estimated

Location

Remote

Remote Work

Remote Friendly