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.