Lead Machine Learning Engineer - Remote

Overview

Wizard is revolutionizing the shopping experience using the power of generative AI and rich messaging technologies to build a personalized shopping assistant for every consumer.

In Short

• Build and maintain a training pipeline for fine-tuning open-source LLMs.
• Serve and Deploy the LLM on GPUs.
• Formulate novel product issues to ML problems.
• Triage customer issues to model/system shortcomings.
• Keep up with developments in the generative AI space.

Requirements

• MS/Ph.D. in computer science, mathematics, or quantitative discipline.
• 5+ years of ML experience.
• Experience developing machine learning-driven products at scale.
• Experience with PyTorch for solving ML problems.
• Understanding of ML systems in production at scale.
• Experience with large data systems and data pipelines.
• Ability to ramp up on tech stack including GCP, Kubernetes, and Python.
• Experience with model deployment using Flask or FastAPI.

Benefits

• Massive growth potential in an early-stage startup.
• Competitive compensation packages including equity.
• Comprehensive medical coverage.
• 401(k) Plan and life & disability insurance.
• Flexible PTO and 12 paid holidays.