Machine Learning Engineer

About the role

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.

See the posting on Hestus, Inc.'s site