Machine Learning Engineer

Location: San Mateo, CA

About Us: We are Hestus, and we build AI-powered CAD to speed up the hardware development process. Our team is small, scrappy, and growing fast. We are looking for a versatile and enthusiastic Machine Learning Engineer to join us. If you enjoy wearing multiple hats and are passionate about technology, we want to hear from you!

Responsibilities:

• Design, train, and evaluate custom machine learning models and embeddings behind our AI-powered CAD.
• Own the entire data pipeline: from data wrangling and curation through to the model training framework.
• Build and improve our training and evaluation infrastructure, including experiment tracking, metrics, and reproducibility.
• Take models from prototype to production: deployment, monitoring, and iteration based on real-world performance.
• Collaborate with cross-functional teams to turn product needs into machine learning solutions and ship new features.
• Write clean, efficient Python, and participate in code reviews, testing, and debugging.
• Stay current with the latest machine learning research (LLMs, embeddings, and representation learning) and bring new ideas to the team.

Requirements:

• Proven experience as a Machine Learning Engineer, with at least 4 years in industry or equivalent academic experience (thesis-based Master’s or PhD).
• Strong proficiency in Python; write clean, efficient code with minimal handholding.
• Demonstrated experience developing novel, custom machine learning models and embeddings: inventing new approaches rather than applying off-the-shelf LLM or image-processing tools.
• Hands-on experience with large language models (LLMs).
• Familiarity with a deep learning framework such as PyTorch, TensorFlow, or JAX.
• Ability to work in a fast-paced startup environment and manage multiple tasks simultaneously.
• Excellent problem-solving skills and attention to detail.
• Strong communication and teamwork skills.

Bonus Skills:

• Experience with ML platforms such as Amazon SageMaker, Azure ML, or Google Cloud AI Platform.
• Experience with cloud platforms (e.g., AWS, Google Cloud).
• Experience with backend development and frameworks (e.g., Django, Flask, SQL).
• CAD, geometry, or computational geometry domain experience.

What We Offer:

• Competitive salary and equity options.
• Comprehensive health insurance, including medical, dental, and vision coverage.
• Catered lunch.
• Unlimited time-off.
• A collaborative and inclusive work environment.
• Opportunities for professional growth and development.
• The chance to be part of a passionate team building innovative products.