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
6+ years
About Draftaid
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

fulltimeMississauga, ON, CA / Mississauga, Ontario, CA / RemoteMechanical$40K - $80K6+ years

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

Sign up to see more ›