Staff Machine Learning Engineer

Description:

Yelp’s mission of connecting people with great local businesses involves leveraging cutting-edge Machine Learning (ML) and Artificial Intelligence (AI) to scale across a vast base of users and businesses. As a Staff-level Machine Learning Engineer on the Content Contributor Intelligence team, you will contribute to building connections across millions of users and business listings. You will work with advanced industry tools, including neural networks (NNs), large language models (LLMs), and embedding techniques for text, images, and videos. Traditional ML methods, such as XGBoost and linear models, will also be used to enhance systems. You will manage the full ML lifecycle—from training models to deploying them in production—and contribute to ML platforms that rely on these models.

This role is fully remote within Canada, offering flexibility in location. We encourage applicants even if they do not meet every requirement listed.

Responsibilities:

• Conduct end-to-end analyses, from data wrangling using SQL or Python to statistical modeling, and presenting business insights.

• Mentor and guide junior engineers, fostering a learning culture.

• Work with large and complex textual and visual datasets.

• Support the development and deployment of machine learning models for both offline, batch-based data products and online, real-time services.

• Work on text and visual understanding with the contributor and visual intelligence team, fine-tuning transformer models for various input types.

• Automate and productionize model pipelines within Python services.

• Advocate for best practices in ML development and operations and mentor newer engineers on those practices.

Qualifications:

• Experience with developing and productionizing machine learning models, especially in neural networks and computer vision, including supported data pipelines.

• Familiarity with ML tools like PyTorch, TensorFlow, Spark MLlib, XGBoost, and Sklearn.

• Strong Python coding skills or equivalent (Java, C++).

• Solid understanding of engineering and infrastructure best practices.

• Curiosity and persistence in uncovering solutions to new problems.

• Experience with LLMs, including using APIs (OpenAI, Bedrock, etc.), prompt engineering, and evaluation is highly valued.

Compensation and Benefits:

• Annual compensation range: $135,000 - $274,000, with potential bonuses and restricted stock units.

• Remote work option available across Canada.

• Comprehensive benefits package (details available on Yelp's website).

Additional Information:

• Yelp is an equal opportunity employer, considering applicants regardless of race, color, religion, sex, national origin, ancestry, age, disability, or other protected status.