ML Engineer

Wizard revolutionizes shopping using generative AI and messaging technologies to build personalized shopping assistants that find the best products across the internet.
$185,000
Machine Learning
Mid-Level Software Engineer
Remote
1+ year of experience
AI · E-Commerce

Description For ML Engineer

Wizard is at the forefront of revolutionizing the e-commerce landscape by leveraging generative AI and advanced messaging technologies. We're building a next-generation shopping assistant that personalizes the entire shopping experience for consumers, scanning the internet to find the perfect products based on individual needs.

We're seeking ML Engineers to join our team, with a focus on NLP expertise. The role involves working on sophisticated dialogue systems, from fine-tuning Large Language Models (LLMs) to building robust data pipelines and entity resolution systems. Even if you don't have direct NLP experience but possess strong Python skills and ML exposure, we welcome your application.

As an ML Engineer at Wizard, you'll be instrumental in building and maintaining training pipelines for LLM fine-tuning, working with various architectures and optimizations. You'll be responsible for GPU deployment of models, ensuring optimal performance and resource utilization. The role requires quick problem-solving abilities and the capacity to translate product challenges into ML solutions.

We offer a competitive compensation package starting at $185,000, along with comprehensive benefits including medical, dental, and vision coverage, mental health support through Spring Health, and generous parental leave. Join us in shaping the future of AI-powered shopping experiences while working with cutting-edge technologies in a fast-paced, growth-oriented environment.

The ideal candidate will have 1-3 years of ML experience, with either advanced degrees in relevant fields or exceptional practical experience. You'll work with modern technologies including GCP, Kubernetes, PyTorch, and Node.js, making this an excellent opportunity for growth in the rapidly evolving field of generative AI.

Last updated 2 months ago

Responsibilities For ML Engineer

  • Build and maintain training pipeline for fine-tuning open-source LLMs
  • Serve and Deploy LLM on GPUs
  • Formulate novel product issues to ML problems and propose solutions
  • Triage customer issues to model/system shortcomings
  • Keep up with generative AI space and incorporate latest developments

Requirements For ML Engineer

Python
Node.js
Kubernetes
  • MS/Ph.D. in computer science, mathematics, or quantitative discipline or exceptional work experience
  • 1-3 years ML Experience
  • Experience developing machine learning-driven products at scale
  • Experience with PyTorch
  • Understanding of ML systems in production at scale
  • Experience with large data systems and 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

Medical Insurance
Dental Insurance
Vision Insurance
Mental Health Assistance
401k
Parental Leave
  • 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

Interested in this job?

Jobs Related To Wizard ML Engineer

ML Engineer

ML Engineer role at Wizard, building AI-powered shopping assistants using LLMs and NLP, offering $185K salary with comprehensive benefits.

Machine Learning Engineer

Machine Learning Engineer role at Peloton focusing on personalization and content recommendations, offering competitive salary and comprehensive benefits in New York City.

Research Engineer - AI Safety

Research Engineer position at Google DeepMind focusing on AI safety, alignment, and risk mitigation for advanced AI systems like Gemini.

Software Engineer - Python (ML)

Python Engineer position at Builder.ai focusing on Machine Learning and AI services, building data pipelines and productionizing ML models in London.

Side Hustle Expert - AI Prompt Engineer

AI Prompt Engineer position focused on developing AI-powered solutions and mentoring community members in side hustle opportunities.