Machine Learning Software Engineer

About Us We're building a co-pilot for hardware designers. Our mission is to enable 9M mechanical engineers to iterate through designs 1000x faster.We are building our geometry + physics driven foundation model for each class of part design. We have line of sight to foundation model for our first physics of interest and are working to expand its capabilities and improve its performanceAlready raised the first round of capital from Khosla Ventures!About YouYou've shipped products in an industry that requires AI to come up with solutions in high dimensional spaces with complex multimodal data. Examples include drug discovery, weather prediction, and bioinformatics where complex fields are key. You are genuinely excited when you hear phrases like low dimensional embeddings, latent spaces, forecasting, prediction, and generalizability.Your ResponsibilitiesDesign, develop, and deploy models to take 3D geometric data combined with other information and predict the output of a physical systemTrain novel transformer like architecturesAssist in the strategy, planning the product roadmap, and prioritize the development in partnership with early customers and design partnersBuild and ship critical product featuresLearn a lot while building products that engineers will love. Also learn about entrepreneurship!Qualifications5+ years of experience developing and shipping products2+ years of experience working on prediction, forecasting, or data generation with different data modalities. Experience architecting, training and deploying a feature in resource constrained environments.Experience with the related technologies, libraries, and languages: Python, C++, PyTorch (3D)/TensorFlow/JAX; plus to have GPU programming Startup experience is a strong advantage.You are excited about entrepreneurship, taking things from 0 to 1, and have a continuous learning mindset. You are excited about building a geometry and physics based model for a vast variety of design problems (e.g. heat dissipation system for chips to landing gears)!You thrive when you have autonomy and ownership over your workLocationWe're located in Palo Alto, CA near Caltrain station. We're backed by Khosla and Eclipse Ventures.Benefits and perksGenerous compensation & equity (stock options)Health + dental + 401kLunch, drinks, and snacks provided for in-person days

Created: 2025-02-09