Machine Learning Research Engineer (Member of Technical Staff)
Perplexity AI
- Berlin, Germany
- Lead/Staff
- 3+ years
- Posted
Skills
Required
- search / recommender systems
- PyTorch
- distributed training techniques
- performance optimization
- search and retrieval systems
- metrics
Preferred
- representation learning
- contrastive learning
- multilingual
- multimodal
- AI/ML
- NeurIPS
- ICML
- ICLR
- ACL
- CVPR
Perplexity, a cutting-edge technology company focused on advanced search technologies. As a Machine Learning Research Engineer, you will play a key role in building and optimizing our search platform and model stack. You will design, train, and optimize large-scale deep learning models, conduct advanced research in representation learning, and collaborate with various teams to ensure high-quality delivery.
Responsibilities
- Architect and build core components of the search platform and model stack, focusing on retrieval and ranking.
- Design, train, and optimize large-scale deep learning models using frameworks like PyTorch, leveraging distributed training and hardware acceleration.
- Conduct advanced research in representation learning, including contrastive learning, multilingual, and multimodal modeling for search and retrieval.
Requirements
- The ideal candidate will have a minimum of 3 years of experience in search or recommender systems, strong proficiency in PyTorch, and a proven track record in large-scale search or recommender systems.
- Minimum of 3 years (preferably 5+) working on search, recommender systems, or closely related research areas
- Strong proficiency with PyTorch, including experience in distributed training techniques and performance optimization for large models
- Proven track record with large-scale search or recommender systems
- Deep understanding of search and retrieval systems, including quality evaluation principles and metrics
- Self-driven, with a strong sense of ownership and execution
Preferred
- Expertise in representation learning, including contrastive learning and embedding space alignment for multilingual and multimodal applications
- Strong publication record in AI/ML conferences or workshops (e.g., NeurIPS, ICML, ICLR, ACL, CVPR, SIGIR)
