Search Machine Learning Engineer Intern (AI)

TL;DR

Search Machine Learning Engineer Intern (AI): Improving search quality, experimenting with new models, and shipping features that directly impact how users search and discover information, with an accent on retrieval and ranking. Focus on training evaluating models for retrieval, ranking, and classification tasks.

Location: In-person in the Belgrade office

Company

Perplexity is looking for a Search Machine Learning Engineer Intern to help build the next generation of advanced search technologies, with a focus on retrieval and ranking.

What you will do

• Contribute to experiments that improve search quality through better models, data usage, and evaluation tools, under the guidance of senior engineers.
• Design and implement components of the search platform and model stack, including retrieval, ranking, and classification models.
• Train evaluating models (including LLM-based approaches) for retrieval, ranking, and classification tasks.
• Support deployment and monitoring of search and ranking models in a scalable and performant way.
• Help build and iterate on RAG pipelines for grounding and answer generation.

Requirements

• Strong foundation in machine learning and statistics, with coursework or projects related to information retrieval, ranking, or recommender systems.
• Experience with Python and common ML frameworks (e.g. PyTorch, TensorFlow, JAX) through academic, open source, or personal projects.
• Self-driven and curious, with a strong sense of ownership, willingness to learn, and comfort working in a fast-paced environment

Nice to have

• Familiarity with evaluating model quality using offline metrics and/or A/B testing is a plus, but not required.
• Previous experience (internships, research, or significant projects) working on search, recommendation, or NLP is a plus, but not required.
• Experience with Rust will be a plus