Staff Machine Learning Engineer

November 30, 2024

Apply Now

Receive Emails with Similar Jobs

Wizard AI

WebsiteLinkedInAll Job Openings

Wizard AI is a company that provides a tailored shopping experience powered by artificial intelligence. It offers a unique text-based service that curates product recommendations across the internet, utilizing AI models that understand user preferences to predict needs and make personalized suggestions. Wizard AI simplifies shopping by finding, ordering, and tracking products, as well as managing returns through SMS communication. This service is designed to save time and improve the convenience of online shopping by efficiently juggling product data, customer reviews, and other digital content to make informed recommendations for customers.

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