ML Engineer
As an ML Engineer at Mach9, you will build perception models for an AI-enabled CAD system, focusing on extracting 3D object and line features from LiDAR point clouds and imagery. This role involves end-to-end ownership from research to production.
About the job
The role At Mach9, ML Engineers build the perception models at the core of our AI-enabled CAD system. We build models to extract 3D object and line features from dense LiDAR point clouds and imagery. Our unique data advantage allows us to develop and train cutting edge 3D scene understanding models that serve real surveyors and engineers in the field.This role is both research-driven and product-focused. You'll design and train the models that power our automated extraction pipeline — image and 3D detection and localization — and work end-to-end from research prototype to production feature. You'll partner closely with infrastructure and product teams to take ideas from a paper to deployed capabilities.This role is ideal for early-to-mid-career ML engineers who thrive on end-to-end ownership and are able to move fluidly from dissecting a new architecture paper to shipping the product feature that the resulting ML model backs.
Responsibilities
- Design, train, and evaluate computer vision and 3D ML models for extracting CAD-grade geometry and features from dense LiDAR and imagery.
- Drive ML research that translates directly into product capabilities: prototyping new approaches, running experiments, and identifying what’s shippable.
- Own models through the full product lifecycle: problem framing, data strategy, training, evaluation, and final integration into our cloud-based CAD software, Digital Surveyor.
- Develop evaluation methodology and metrics that reflect real surveying and engineering accuracy requirements.
- Work with ML infrastructure engineers to scale training and inference of your models and with product teams to align your model’s behavior with what the user wants.
Requirements
- Master's or PhD in Machine Learning, Computer Vision, Computer Science, or a related field, or equivalent industry experience.
- Strong foundation in computer vision and deep learning, with hands-on experience training models for segmentation, detection, or 3D understanding.
- Experience taking a ML model from research/prototype to production, not just publishing or benchmarking.
- Working knowledge of geometric concepts relevant to 3D perception like coordinate systems and 3D transforms.
- Strong communication skills and the ability to collaborate with researchers, other engineers and product stakeholders.
- Proficient with Python and a production-quality ML library like PyTorch, JAX, or TensorFlow.
Bonus qualifications
- Experience with common 3D deep learning architectures, like point cloud backbones such as PTv3, sparse convolutions, or 3D detection/segmentation networks.
- Experience with large unstructured datasets — imagery and 3D point clouds — at scale.
- Experience delivering production-grade models with optimization techniques such as quantization, pruning, distillation, or runtime acceleration (e.g., TensorRT, ONNX Runtime).
- Familiarity with multi-GPU training and experiment management (Weights & Biases or similar).
- Publications or strong open-source contributions in computer vision or 3D machine learning.
Skills
Computer Vision, Deep Learning, Python, PyTorch, JAX, TensorFlow, 3D Deep Learning, Lidar, Quantization, TensorRT
Similar jobs
ML Engineering jobs
Thinking Machines Lab
San Francisco, CA
Build and operate the engineering systems that support post-training research, including reinforcement learning infrastructure, sandboxed execution, data pipelines, and agent scaffolding. The role requires strong Python and systems engineering skills, project ownership, and a relevant bachelor’s degree or equivalent experience.
OpenAI
San Francisco, CA
Build research infrastructure and tooling that enables AI models to design silicon, including reinforcement learning environments, EDA integrations, evaluations, and experiment workflows. The role requires strong software engineering fundamentals and comfort working across research, tooling, and chip-design systems.
Rollstack
United StatesBuild production AI capabilities for automated slide and document generation, working across LLM applications, data analysis, and content generation. The role requires 3+ years in machine learning and NLP, advanced Python, and experience with LLM frameworks and production systems.
ClickUp
United States
Build and operate large-scale ranking and retrieval systems that power search relevance, including hybrid lexical/vector search, embeddings, query understanding, and permission-aware retrieval. Requires a bachelor's degree and 5+ years of ML engineering experience in ranking or information retrieval.
PathAI
Boston, MADevelop and deploy machine learning models for biomedical research and AI products, collaborating with scientific, engineering, and product teams. Requires an advanced quantitative degree, substantial ML experience, Python proficiency, and experience bringing models into production or research applications.