Internship - Machine Learning Research Engineer

The role

This person will push search quality forward through models, data, and tools. They will train and optimize large-scale deep learning models and conduct research in representation learning.

What you will do

• 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
• Leverage distributed training and hardware acceleration with a focus on retrieval and ranking models
• Conduct research in representation learning including contrastive learning, multilingual, evaluation, and multimodal modeling
• Build and optimize RAG pipelines for grounding and answer generation

Required 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
• Publication record in AI/ML conferences or workshops (e.g., NeurIPS, ICML, ICLR, ACL, EMNLP, SIGIR)

About the company

perplexity