Remote Machine Learning Engineer III

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.

Benefits:

• 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.