Machine Learning Engineer

Department
:
Information Technology,
Type: Full Time

Job Title: Machine Learning Engineer / Research Engineer

Pay: $$110,000 – $165,000 Base Salary + Equity

Shift: N/A

Location: San Mateo, CA (Peninsula) – Onsite Preferred

Schedule: Full time, Permanent Role

Visa Sponsorship: Not Available

Relocation Assistance: Not Available

We are looking for a highly skilled Machine Learning Engineer / Research Engineer to join our founding team and help develop intelligent systems that transform how hardware and mechanical engineers design products. This is a unique opportunity to work at the intersection of cutting‑edge machine learning research and real‑world engineering applications. You'll collaborate directly with founders, engineers, and customers to design, train, deploy, and continuously improve machine learning systems that accelerate CAD workflows and hardware design.

As one of the earliest ML hires, you will have significant ownership over technical direction, architecture decisions, and the long‑term evolution of our AI platform.

• Design, train, and optimize custom deep learning models that understand CAD workflows and generate intelligent next‑step design recommendations.
• Develop novel machine learning approaches for geometry, design, and engineering‑related datasets.
• Evaluate emerging research in areas such as sequence modeling, geometric deep learning, representation learning, and foundation models.

• Build and maintain scalable Python‑based training, evaluation, and experimentation pipelines.
• Transform complex, real‑world CAD and geometry data into high‑quality training datasets and signals.
• Implement robust offline and online evaluation frameworks to measure model performance and business impact.

• Own the complete ML lifecycle from research and prototyping through deployment, monitoring, and optimization.
• Architect model‑serving infrastructure and backend components that enable fast, reliable integration into CAD environments.
• Establish best practices for experimentation, logging, model versioning, and performance monitoring.

• Work closely with founders, mechanical engineers, hardware engineers, and early customers to understand workflows and translate them into ML solutions.
• Collaborate with backend engineers on APIs, infrastructure, data models, and platform scalability.
• Help define the long‑term strategy for applying machine learning to hardware and CAD design.

• 4+ years of hands‑on machine learning experience in industry, research, or a combination of both.
• Equivalent Master's or PhD research experience will be considered.
• Demonstrated success designing, training, improving, and deploying machine learning models—not simply utilizing hosted AI APIs.

• Expert‑level proficiency with PyTorch (preferred) or similar frameworks such as Tensor Flow or JAX.
• Experience implementing custom architectures, loss functions, optimization methods, and training loops.
• Strong understanding of model evaluation, experimentation, and performance trade‑offs.

• Strong Python programming skills with experience building production‑ready systems.
• Ability to write clean, maintainable, and well‑tested code with appropriate documentation and abstractions.
• Experience developing scalable ML infrastructure and backend services.

• Proven ability to independently drive projects from concept through deployment.
• Experience building end‑to‑end ML systems including data pipelines, experimentation frameworks, model training, deployment, and monitoring.
• Comfortable solving ambiguous, open‑ended technical problems.

• Excellent communication skills with the ability to explain technical concepts to both technical and non‑technical stakeholders.
• Experience working cross‑functionally with engineers, product teams, researchers, and customers.

• Thrives in fast‑paced, high‑ownership environments.
• Comfortable wearing multiple hats across machine learning, research, backend…