Description:

Wizard is seeking a Lead Machine Learning Engineer to work remotely in the USA, focusing on dialog systems and generative AI technologies.
The role involves building and maintaining a training pipeline for fine-tuning open-source LLMs, testing various LLM architectures, and deploying models on GPUs.
Responsibilities include formulating product issues into ML problems, triaging customer issues, and keeping up with developments in the generative AI space.
Candidates with a background in NLP are preferred, but those with strong Python skills and a willingness to learn are also encouraged to apply.

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.
A minimum of 5 years of experience in machine learning is necessary.
Candidates must have experience developing machine learning-driven products at scale and solving problems using Machine Learning with PyTorch.
An exceptional understanding of designing ML systems in production at scale and identifying performance bottlenecks is essential.
Practical experience with large data systems, data models, and both batch and streaming data pipelines is required.
Candidates should be able to quickly ramp up on the tech stack, which includes GCP, Kubernetes, Airflow, Pandas, PyTorch, Python, and Node.js.
Extensive experience with model deployment using frameworks like Flask or FastAPI is necessary.

The position offers the opportunity to be part of an early-stage startup with significant growth potential.
Competitive compensation packages are provided, including equity.
Health benefits include comprehensive medical coverage, dental and vision insurance, OneMedical memberships, mental healthcare through the Spring health platform, eyewear benefits, and a Rightway Health Guide.
Wealth benefits include a 401(k) plan and life and disability insurance covered by Wizard.
Work/life balance is supported with flexible PTO, sick time, 12 paid holidays, and 16 weeks of parental leave for primary and secondary caregivers.