What you'll do

The Senior Machine Learning Engineer will be responsible for architecting and building core components of the search platform and model stack. This will include designing, training, and optimizing 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.

• Relentlessly push search quality forward — through models, data, tools, or any other leverage available
• Architect and build core components of the search platform and model stack
• Design, 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

What you need

The successful candidate will need to have a deep understanding of search and retrieval systems, including quality evaluation principles and metrics. They will also need to have a proven track record with large-scale search or recommender systems, and strong proficiency with PyTorch, including experience in distributed training techniques and performance optimization for large models.

XML job scraping automation by YubHub