Staff Machine Learning Engineer

November 30

Apply Now

Receive Emails with Similar Jobs

Wizard AI

WebsiteLinkedInAll Job Openings

51 - 200 employees

🤖 Artificial Intelligence

🛍️ eCommerce

🛒 Retail

💰 $50M Series A on 2021-10

Description

• 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. • Wizard is hiring ML Engineers, preferably with a background in NLP, to work on all aspects of dialog systems, from fine-tuning LLMs to serving models and building data pipelines and entity resolutions. • Build and maintain a training pipeline for fine-tuning open-source LLMs. • Serve and Deploy the LLM on GPUs. • Triage customer issues to model/system shortcomings and communicate the complex architecture in a simple, comprehensible way.

Requirements

• MS/Ph.D. in computer science, mathematics, or another quantitative discipline or exceptional work experience in ML engineering • 5+ years ML Experience • Experience developing machine learning-driven products at scale • Experience solving problems using Machine Learning with PyTorch • Exceptional understanding of designing ML systems in production at scale and applying a quantitative approach to identifying performance bottlenecks and cost/performance tradeoffs • Practical experience with large data systems, data models, and batch and streaming data pipelines • Ability to ramp up quickly on our tech stack, which includes GCP, Kubernetes, Airflow, Pandas, PyTorch, Python, and Node.js • Extensive experience with model deployment, both static within applications and dynamic, using Flask, FastAPI, or similar frameworks

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