ML Engineer
Description For ML Engineer
Wizard is at the forefront of transforming the e-commerce landscape by leveraging generative AI and messaging technologies to create personalized shopping assistants. As an ML Engineer at Wizard, you'll be integral to developing sophisticated dialog systems and fine-tuning LLMs. The role combines deep technical expertise in machine learning with practical application in a fast-paced startup environment.
You'll work on cutting-edge problems in NLP, from fine-tuning open-source LLMs to building robust data pipelines and entity resolutions. The position requires strong Python skills and machine learning expertise, though candidates with strong programming backgrounds and a willingness to learn ML are also encouraged to apply.
The role offers significant growth potential in an early-stage startup, with competitive compensation including a $210,000 salary and equity. Wizard provides comprehensive benefits including full medical coverage, dental and vision insurance, mental health support through Spring Health, and generous parental leave policies.
This is an excellent opportunity for experienced ML engineers who want to shape the future of AI-powered shopping experiences. You'll work with modern technologies including PyTorch, GCP, and Kubernetes, while staying at the forefront of the rapidly evolving generative AI landscape. The role combines technical depth with practical impact, making it ideal for those who want to build products that directly enhance consumer shopping experiences.
Responsibilities For ML Engineer
- 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 and propose solutions
- Triage customer issues to model/system shortcomings
- Keep up with the fast-growing generative AI space
Requirements For ML Engineer
- MS/Ph.D. in computer science, mathematics, or another quantitative discipline or exceptional work experience
- 3+ years ML Experience
- Experience developing machine learning-driven products at scale
- Experience solving problems using Machine Learning with PyTorch
- Understanding of designing ML systems in production at scale
- Experience with large data systems and data pipelines
- Ability to work with GCP, Kubernetes, Airflow, Pandas, PyTorch, Python, and Node.js
- Experience with model deployment using Flask, FastAPI, or similar frameworks
Benefits For ML Engineer
- Early-stage startup with massive growth potential
- Competitive compensation packages, including equity
- Comprehensive medical coverage
- Dental & vision insurance
- OneMedical memberships for employees and dependents
- Spring health platform for mental healthcare
- XP Health eyewear benefits ($180, 3x per year)
- Rightway Health Guide
- 401(k) Plan
- Life & Disability insurance
- Flexible PTO and sick time
- 12 paid holidays
- 16 weeks parental leave for primary and secondary caregivers
Jobs Related To Wizard ML Engineer
