Senior ML Engineer at Draftaid (W24)
$140K - $200K  •  0.25% - 1.50%
Go from 3D models to CAD drawings using AI
Mississauga, ON, CA / Mississauga, Ontario, CA
Full-time
US citizenship/visa not required
6+ years
About Draftaid

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.

About the role
Skills: Torch/PyTorch, Python, TensorFlow, PostgreSQL

About Us

  • 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.

Responsibilities

  • 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

Requirements

  • 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

Preferred Qualifications

  • 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

What We Offer

  • 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

Other jobs at Draftaid

fulltimeMississauga, ON, CA / Mississauga, Ontario, CAMachine learning$140K - $200K0.25% - 1.50%6+ years

fulltimeMississauga, ON, CA / Mississauga, Ontario, CAFull stack$140K - $200K0.25% - 1.50%6+ years

Hundreds of YC startups are hiring on Work at a Startup.

Sign up to see more ›