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 25 days 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

Software Engineer 2- AI Full Stack Development

Mid-level Software Engineer position at Microsoft focusing on AI and full-stack development, requiring 4+ years of experience and expertise in AI/ML technologies.

Research Product Manager, Google Cloud, Domain Applied ML

Lead AI/ML research product management at Google Cloud, driving innovation in Generative AI and machine learning infrastructure.

Research Scientist, Market Algorithms

Research Scientist position at Google focusing on market algorithms, combining ML, economics, and computer science research with practical applications.

Research Scientist

Research Scientist position at Google Research focusing on machine learning and AI systems development, requiring PhD and programming expertise.

Software Developer III, Machine Learning, Google Cloud AI

Software Developer III position at Google Cloud AI, focusing on machine learning and AI development with opportunities to work on large-scale systems impacting billions of users.