AI/ML Engineer for Advanced CAE & Generative Design

Stealth Startup; San Francisco (CA); On-site; USD 150,000 - 210,000; Full time

Job summary

Stealth Startup in San Francisco seeks an AI/ML Engineer to design, develop, and deploy ML models that accelerate engineering simulations and design exploration. You will integrate ML into high-fidelity CAE pipelines and collaborate with CAE specialists and AI researchers to push the boundaries of computational design.

You’ll lead projects from research to production, applying time-series, meshes, and physics-informed data techniques, and stay current with AI advances to deliver production-ready

Qualifications

• 5+ years of experience developing and deploying AI/ML models in applied engineering or scientific domains.
• 2+ years of technical leadership experience guiding AI/ML projects from research to production.
• Strong proficiency in Python and modern AI/ML frameworks such as PyTorch, JAX, or TensorFlow.
• Experience developing AI/ML solutions for sequential, spatial, or physics-informed data, including time series, text, meshes, or simulation outputs.
• Familiarity with MLOps practices and experience building AI/ML systems end-to-end, from prototyping through scalable production deployment.
• Excellent communication and collaboration skills, with the ability to work effectively with software engineers, CAE specialists, and applied scientists in a distributed, interdisciplinary environment.

Responsibilities

• Research, design, and develop AI/ML algorithms to solve complex challenges in computer-aided engineering, simulation, and design automation.
• Lead model and system performance improvements, including optimizing accuracy, analyzing outputs, and addressing system-level bottlenecks.
• Stay current with advances in AI and simulation research, bringing emerging techniques in machine learning, generative design, and physics-informed modeling into production-ready solutions.
• Build and optimize AI-powered solutions such as adaptive solvers, reduced-order models, generative design systems, and real-time validation tools.
• Integrate machine learning models into high-fidelity engineering simulation pipelines.
• Develop intelligent, data-driven models that enable faster failure detection, prediction, and diagnosis.
• Contribute to the development of next-generation CAE capabilities powered by AI.
• Help build foundational AI systems that transform how engineering simulations are designed, executed, and analyzed.

Skills

Python

PyTorch

JAX

TensorFlow

MLOps

AI/ML in engineering

Leadership

Communication

Job description

Stealth Startup in San Francisco seeks an AI/ML Engineer to design, develop, and deploy ML models that accelerate engineering simulations and design exploration. You will integrate ML into high-fidelity CAE pipelines and collaborate with CAE specialists and AI researchers to push the boundaries of computational design.

You’ll lead projects from research to production, applying time-series, meshes, and physics-informed data techniques, and stay current with AI advances to deliver production-ready