Jobs · Information Technology

Machine Learning Engineer - Relevance & Learning Systems

Wizard AI · United States · 3 mo ago
RemoteRemoteInformation Technology$225k–$280k/yrFull-time

About the role

We’re looking for a Machine Learning Engineer to design and build feedback driven learning systems that improve our AI agent over time. This is not a traditional RL research role, we’re focused on building systems that learn from real user behavior and improve production. You’ll be working at the intersection of a live conversational agent and real shopping behavior – the feedback signal quality here is unusually rich compared to traditional search.

What You’ll Do

  • Build and productionize feedback loops that improve agent performance over time
  • Build the evaluation infrastructure – offline metrics, regression suites, and experiment analysis
  • Own the signal pipelines end-to-end: instrument events, build clean labeled datasets, and translate user behaviors into reliable learning signals
  • Design lightweight reinforcement learning / bandit-style approaches where appropriate
  • Partner closely with product and engineering to define success metrics and optimize for them
  • Design and analyze experiments that validate whether learning system changes actually improve real outcomes
  • Improve ranking, recommendations and decision making within the agent
  • Iterate quickly: Ship → measure → learn → improve

What Success Looks like

  • You ship quickly and drive measurable improvements in core product metrics
  • You turn noisy user behavior into reliable learning signals that improve the agent over time
  • You own systems end to end and operate comfortably in production

Ideal Background

  • 5-8 years hands on experience building and shipping ML systems
  • Bachelor’s or Master's degree in computer science
  • Experience shipping ML systems to production and have worked on recommendation systems, ranking, personalization or optimization problems
  • Deep knowledge in Python and model ML tooling
  • Pragmatic: you choose simple, effective solutions over theoretically perfect ones

Compensation & Benefits

  • The expected base salary range for this role is $225,000 - $280,000 USD, and will vary based on skills, experience, role level, and geographic location. Final compensation will be determined by considering these factors alongside overall role scope and responsibilities.
  • In addition to base salary, Wizard offers:
    • Equity in the form of stock options
    • Medical, dental, and vision coverage
    • 401(k) plan
    • Flexible PTO and company holidays
    • Fully remote work within the United States
    • Periodic company offsites and team gatherings

Wizard is committed to fair, transparent, and competitive compensation practices.

Similar jobs