Principal Machine Learning Engineer / Team Lead

About Us

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. We scour the entire internet of products and ratings across brands and retailers to find the best products for every consumer’s personalized needs. Using an effortless text-based interface, Wizard AI is always just a text away. The future of shopping is here. Shop smarter with Wizard.

The Role

Wizard is seeking a Principal Machine Learning Engineer and Team Lead to join our rapidly growing AI team. This role will drive the research, development and implementation of models and algorithms.  The role extends beyond traditional boundaries, requiring the candidate to innovate in model optimization, system scalability, and the integration of cutting-edge AI technologies to redefine conversational experiences. As a engineer and team lead, you will set the technical direction and mentor a team of ML engineers, influence our AI strategy, and collaborate with cross-functional teams to drive product innovation and excellence.

Key Responsibilities:

• Architect and oversee development for novel product issues to ML problems and propose and iterate on solutions quickly. We value the fast pace of development.
• Triage customer issues to model/system shortcomings and communicate the complex architecture in a simple, comprehensible way to internal stakeholders
• Build and maintain pipelines for fine-tuning open-source LLMs. Testing different architectures at different sizes and quantizations is important.
• Serve and Deploy the LLM on GPUs
• Working with Product and Engineering counterparts, translate customer needs and business requirements into long term vision and roadmap at Wizard
• Act as a thought leader within the company, staying abreast of the latest advancements in the Generative AI field and integrating innovative technologies into our systems to maintain a competitive edge

You

• Ph.D.in Computer Science, Mathematics, or another quantitative field or outstanding professional experience in machine learning engineering.
• Over 5 years of industry experience in machine learning, with a proven track record of developing ML-driven products at scale.
• Experience building recommender systems, preferably in the e-commerce space.
• Demonstrated expertise in tackling complex problems using machine learning frameworks like PyTorch.
• A comprehensive understanding of ML system design in production environments, including a quantitative approach to evaluating performance bottlenecks and cost/performance trade-offs.
• Hands-on experience with large-scale data systems, sophisticated data models, and batch and streaming data pipelines.
• Quick adaptability to our technology stack encompasses GCP, Kubernetes, Airflow, Pandas, PyTorch, Python, and Node.js.
• Deep experience in model deployment strategies, utilizing both in-application static methods and dynamic approaches with frameworks like Flask, FastAPI, or equivalents.

Please note you will only be considered for the position if you meet the minimum technical requirements. We offer a remote-friendly environment; however, employees must reside within the United States and be eligible to obtain or hold the legal right to work in this country.

The expected salary for this role is $285,000 - $315,000 depending on skills and experiences. 

Benefits

• Early-stage startup with massive growth potential and ability to grow as Wizard grows
• Competitive compensation packages, including equity
• Health 
  • Comprehensive, high-quality medical coverage
  • Dental & vision insurance
  • OneMedical memberships for you and dependents
  • Spring health platform for mental healthcare personalized to your needs
  • XP Health eyewear benefits ($180, 3x per year)
  • Rightway Health Guide
• Wealth 
  • 401(k) Plan
  • Life & Disability insurance covered by Wizard
• Work/Life
  • Flexible PTO and sick time to take care of yourself and your family
  • 12 paid holidays
  • 16 weeks parental leave for primary and secondary caregivers

Salary : $285,000 - $315,000