Internship - Machine Learning Research Engineer
Join Perplexity AI as a Machine Learning Research Engineer intern to enhance search quality through advanced deep learning techniques.
Posted by employer 1 week ago
First seen on Joblaze 1 week ago
Last verified on the company career page 1 day ago
Skills & Technologies
Not disclosed in this posting: compensation, years of experience, visa sponsorship.
Joblaze summary
In this internship, the individual will focus on enhancing search quality by training and optimizing large-scale deep learning models, particularly in retrieval and ranking. Proficiency in PyTorch and experience with distributed training techniques are essential, along with a solid understanding of search and retrieval systems. This role is ideal for candidates with a background in AI/ML research, especially those who have published in relevant conferences. The position offers a hands-on opportunity to contribute to innovative projects within a dynamic team at Perplexity AI.
Joblaze insights
- Listed 9 days ago on Joblaze — tracked directly from Perplexity AI's career page.
- Perplexity AI has 115 other open roles including Member of Technical Staff (Applied AI Engineer, Agent Capabilities), Strategic Business Operations Lead, API, Technical Customer Success Manager, API.
- 481 active PyTorch roles on Joblaze right now.
- 20 active DeepSpeed roles on Joblaze right now.
- 10 active FSDP roles on Joblaze right now.
- Typical PyTorch salary we track runs $150,000–$218,625 (median $188,792, across 248 roles).
Quick facts
- Is the Internship - Machine Learning Research Engineer role remote?
- No — this is an on-site role in Berlin.
- Where is the role based?
- Perplexity AI is hiring for this position in Berlin.
- What's the tech stack?
- Joblaze extracted these technologies from the posting: DeepSpeed, FSDP, PyTorch, RAG pipelines, contrastive learning, multilingual.
- What seniority level is this role?
- Perplexity AI targets intern candidates for this position.
- Is this full-time or contract?
- Internship for this Internship - Machine Learning Research Engineer role at Perplexity AI.
From the original posting
Internship Program Berlin
Internship program: 12 - 24 weeks, full-time, in-person in the Berlin office.
Responsibilities
Relentlessly push search quality forward — through models, data, tools, or any other leverage available.
Train, and optimize large-scale deep learning models using frameworks like PyTorch, leveraging distributed training (e.g., PyTorch Distributed, DeepSpeed, FSDP) and hardware acceleration, with a focus on retrieval and ranking models.
Conduct research in representation learning, including contrastive learning, multilingual, evaluation, and multimodal modeling for search and retrieval.
Build and optimize RAG pipelines for grounding and answer generation.
Qualifications
Understanding of search and retrieval systems, including quality evaluation principles and metrics.
Strong proficiency with PyTorch, including experience in distributed training techniques and performance optimization for large models.
Interested in representation learning, including contrastive learning, dense & sparse vector representations, representation fusion, cross-lingual representation alignment, training data optimization and robust evaluation.
Publication record in AI/ML conferences or workshops (e.g., NeurIPS, ICML, ICLR, ACL, EMNLP, SIGIR).