Principal Machine Learning Engineer - Team Lead

6 days ago

Apply Now

Receive Emails with Similar Jobs

Wizard AI

WebsiteLinkedInAll Job Openings

51 - 200

💰 $50M Series A on 2021-10

Description

• About Us: Wizard is revolutionizing the shopping experience using the power of generative AI and rich messaging technologies. • The Role: Seeking a Principal Machine Learning Engineer and Team Lead to join our rapidly growing AI team. • Key Responsibilities: Architect and oversee development of novel product issues to ML problems. • Triage customer issues to model/system shortcomings and communicate architecture simply. • Build and maintain pipelines for fine-tuning open-source LLMs. • Serve and Deploy the LLM on GPUs. • Collaborate with Product and Engineering to translate needs into vision. • Act as a thought leader, integrating innovative technologies into our systems.

Requirements

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

Benefits

• Early-stage startup with massive growth potential and ability to grow as Wizard grows • Competitive compensation packages, including equity • 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 • 401(k) Plan • Life & Disability insurance covered by Wizard • 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