Description:

Wizard is seeking a Machine Learning Engineer III to work on dialog systems, focusing on fine-tuning LLMs, serving models, and building data pipelines and entity resolutions.
The role involves building and maintaining a training pipeline for fine-tuning open-source LLMs, testing different LLM architectures, and deploying the LLM on GPUs.
Responsibilities include formulating novel product issues to ML problems, triaging customer issues related to model/system shortcomings, and communicating complex architecture simply to internal stakeholders.
The candidate should stay updated with the fast-growing generative AI space and incorporate the latest developments into the system.

Requirements:

A Master's or Ph.D. in computer science, mathematics, or a related quantitative discipline, or exceptional work experience in ML engineering is required.
The candidate should have 1-3 years of experience in machine learning.
Experience in developing machine learning-driven products at scale is necessary.
Proficiency in solving problems using Machine Learning with PyTorch is required.
An exceptional understanding of designing ML systems in production at scale and applying a quantitative approach to identify performance bottlenecks and cost/performance tradeoffs is essential.
Practical experience with large data systems, data models, and both batch and streaming data pipelines is needed.
The ability to quickly ramp up on the tech stack, which includes GCP, Kubernetes, Airflow, Pandas, PyTorch, Python, and Node.js, is important.
Extensive experience with model deployment, both static within applications and dynamic using frameworks like Flask or FastAPI, is required.

The position offers the opportunity to be part of an early-stage startup with massive growth potential, allowing for personal growth as Wizard expands.
Competitive compensation packages are provided, including equity.
Health benefits include comprehensive medical coverage, dental and vision insurance, OneMedical memberships for employees and dependents, a personalized mental healthcare platform, eyewear benefits, and a health guide.
Wealth benefits include a 401(k) plan and life & disability insurance covered by Wizard.
Work/life benefits feature flexible PTO and sick time, 12 paid holidays, and 16 weeks of parental leave for primary and secondary caregivers.