Machine Learning Engineer

ElastixAI INC.; Seattle (WA); On-site; Full time

Benefits offered by this job

Competitive compensation

Startup equity package

Comprehensive medical, dental, and vision coverage

Flexible Time Off (FTO)

Paid parental leave

Gym or fitness benefit

Commuter benefit

Investment in employee learning & development

Job summary

An innovative AI startup is seeking a talented Machine Learning Engineer to play a key role in building their core AI inference platform in Seattle. Responsibilities include designing and developing components, researching and implementing advanced ML techniques, and collaborating with a multi-disciplinary team. The ideal candidate holds a PhD/MS in Computer Science with 3+ years of experience in machine learning, strong programming skills, and the ability to thrive in a dynamic startup environment. Competitive compensation and comprehensive benefits are offered.

Qualifications

• PhD/MS in Computer Science, Computer Engineering, or a related field.
• 3+ years of machine learning R&D and deployment experience.
• Strong proficiency in one or more programming languages such as Python or C++.
• Proficiency in at least one ML framework (e.g., PyTorch, TensorFlow, JAX).
• Solid understanding of software engineering best practices.

Responsibilities

• Design, develop, and maintain core components of the ML platform.
• Research and implement advanced ML techniques to optimize performance.
• Collaborate with engineers for efficient use of hardware resources.
• Contribute to the design of APIs and tools for inference solutions.

Skills

Machine learning R&D

Programming in Python or C++

Collaboration in cross-functional teams

Problem-solving

Understanding of software engineering best practices

Education

PhD/MS in Computer Science or related field

Tools

PyTorch

TensorFlow

JAX

Docker

Kubernetes

AWS

GCP

Azure

Job description

About Elastix AI

We are building the next-gen AI inference platform.

Description

Location: Seattle, WA (Hybrid - 3 days/week in office)

About ElastixAI:

ElastixAI is an early-stage startup poised to revolutionize AI inference infrastructure. We are developing a cutting-edge AI inference solution that dramatically improves efficiency. Our solution is going to dynamically adapt to any deployment, constantly evolving to power the next generation AI use cases.

Role Summary:

We are looking for a talented Machine Learning Engineer to play a key role in building our core AI inference platform. You will be responsible for designing and developing critical components, including ML model deployment, innovative model optimization pipelines, and performance benchmarking and simulation workflows. This is a highly interdisciplinary role where you'll collaborate closely with our multi-disciplinary team to ensure our entire stack works in harmony to deliver optimal AI inference solutions.

Key Responsibilities:

• Design, develop, and maintain core components of our ML platform, focusing on scalability, reliability, and ease of use.

• Research, prototype, and implement advanced ML techniques to optimize inference performance across diverse hardware targets.

• Collaborate with systems and cloud engineers to ensure efficient utilization of underlying hardware resources.

• Contribute to the design of APIs and tools that enable seamless integration and management of our inference solutions.

Required Qualifications:

• PhD/MS in Computer Science, Computer Engineering, or a related field.

• 3+ years of machine learning R&D and deployment experience.

• Strong proficiency in one or more programming languages such as Python, or C++.

• Proficiency in at least one ML framework (e.g., PyTorch, TensorFlow, JAX).

• Solid understanding of software engineering best practices, including data structures, algorithms, and testing.

• Excellent problem-solving abilities and a knack for tackling complex technical challenges.

• Strong communication skills and a proven ability to collaborate effectively in a cross-functional team environment.

• Ability to thrive in a fast-paced, dynamic startup environment.

Preferred/Bonus Qualifications:

• Experience with training generative machine learning models.

• Experience with optimizing the performance of machine learning models.

• Experience leading research initiatives in Machine Learning, Natural Language Processing, and related fields.

• Familiarity with performance analysis, profiling, and optimization techniques.

• Experience with cloud platforms (AWS, GCP, Azure).

• Experience with containerization and orchestration technologies (e.g., Docker, Kubernetes).

What We Offer:

• A chance to be a foundational engineer in an innovative AI startup.

• A dynamic and collaborative work environment and the change to have a significant impact on new technology

• The opportunity to work on challenging problems at the intersection of ML, software, and systems.

• Competitive compensation and startup equity package

• Comprehensive medical, dental, and vision coverage (100% paid by employer)

• Flexible Time Off (FTO)

• Paid parental leave

• Gym or fitness benefit

• Commuter benefit

• Investment in employee learning & development