AI/ML Engineer

About the Role Join an early-stage, venture-backed AI startup reimagining computer-aided engineering (CAE) for the AI era. As an AI/ML Engineer, you'll shape the future of engineering automation by building AI that thinks like an engineer — designing, training, and deploying models that accelerate simulation, improve accuracy, and unlock entirely new ways of exploring the design space. From adaptive solvers and reduced-order modeling to generative design and real-time validation, your work will push the boundaries of what CAE can achieve. You'll collaborate closely with experts in applied AI and mechanical engineering, integrating machine learning directly into high-fidelity simulation pipelines to help customers iterate faster, diagnose failures earlier, and bring better products to market. This is a rare opportunity to join early and make a foundational impact. What You'll Do Research, design, and develop AI/ML algorithms to tackle complex challenges in simulation, design automation, and computer-aided engineering. Build scalable AI solutions and collaborate with cross-functional teams to integrate models seamlessly into existing CAE workflows and infrastructure. Lead performance improvements — from optimizing model accuracy to analyzing outputs and addressing system-level bottlenecks. Guide AI/ML projects from research through to production deployment. Contribute to MLOps practices and help build robust, end-to-end AI/ML systems. What We're Looking For Required: 4+ years of experience developing and deploying AI/ML models with a proven track record of delivering impact in applied engineering or scientific domains. 2+ years of technical leadership guiding AI/ML projects from research to production. Proficiency in Python and modern AI/ML frameworks (e.g., PyTorch, JAX, TensorFlow). Experience with AI/ML algorithms for sequential, spatial, or physics-informed data (e.g., time series, mesh data, or simulation outputs). Familiarity with MLOps practices and building AI/ML systems end-to-end, from prototyping to scalable deployment. Strong understanding of engineering principles in simulation, modeling, and optimization. Bachelor's or advanced degree in Computer Science, Mechanical Engineering, Electrical Engineering, or a related field. Nice to Have: Experience with CAE, CAD, and/or PLM tools (e.g., Ansys, Abaqus, COMSOL, SolidWorks, NX). Experience in a fast-paced startup environment. Compensation & Benefits Salary: $150,000 – $250,000 USD annually Visa sponsorship available Location On-site in San Francisco, CA, United States . Candidates must be available to work from the San Francisco office.