ML Platform Engineer — Inference & Optimization (Hybrid)

ElastixAI INC.; Seattle (WA); Hybrid; 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

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.