Lead Machine Learning Engineer

Promote Project; New York (NY); Remote; USD 70,000 - 120,000; Full time

Job summary

An innovative startup is seeking ML Engineers to shape the future of shopping through generative AI. In this exciting role, you will work on dialog systems, fine-tuning large language models, and building robust data pipelines. You will be part of a fast-paced environment where your contributions will directly impact the shopping experience for consumers. With a focus on personal growth alongside the company's expansion, this position offers a unique opportunity to advance your career in a rapidly evolving field. If you're passionate about machine learning and eager to make a difference, this role is for you.

Benefits

Equity options

Comprehensive medical coverage

Dental & vision insurance

Flexible PTO

12 paid holidays

16 weeks parental leave

Qualifications

• 5+ years of ML experience with a focus on NLP and dialog systems.
• Practical experience with large data systems and ML product development.

Responsibilities

• Build and maintain training pipelines for fine-tuning LLMs.
• Serve and deploy models on GPUs while managing configurations.

Skills

Machine Learning

Natural Language Processing (NLP)

Python

PyTorch

Data Pipeline Development

Model Deployment

Education

MS/Ph.D. in Computer Science or Mathematics

Exceptional work experience in ML engineering

Tools

GCP

Kubernetes

Airflow

Flask

FastAPI

Pandas

Node.js

Job 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. We scour the entire internet of products and ratings across brands and retailers to find the best products for every consumer’s personalized needs. Using an effortless text-based interface, Wizard AI is always just a text away. The future of shopping is here. Shop smarter with Wizard.

The Role

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. If you don’t have an NLP background but have deep experience with Python and some exposure to ML and you are excited about the space and willing to learn, we would still encourage you to apply.

Key Responsibilities:

• Build and maintain a training pipeline for fine-tuning open-source LLMs. Testing different LLM architectures at different sizes and quantizations is important.
• Serve and Deploy the LLM on GPUs. It is important to know different configurations ahead of time to ensure we don’t OOM at runtime.
• Formulate novel product issues to ML problems and propose and iterate on solutions quickly. We value the fast pace of development.
• Triage customer issues to model/system shortcomings and communicate the complex architecture in a simple, comprehensible way to internal stakeholders.
• Keep up with the fast-growing generative AI space and incorporate the latest developments into our system.

You

• 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.
• Health
  • 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.
• Wealth
  • 401(k) Plan.
  • Life & Disability insurance covered by Wizard.
• Work/Life
  • 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.

Location

New York City, New York, United States

Salary

$70,000 - $120,000 a year (US Dollars)

Job Type

Remote job

Tags

• python
• training
• testing
• growth
• lead
• health
• healthcare
• engineer