Member of Technical Staff (Machine Learning Research Engineer)

Responsibilities

We are representing a confidential AI organization seeking a Machine Learning Research Engineer to advance their next-generation search technologies, with particular emphasis on retrieval and ranking systems.

In this role, you will:

• Drive search quality improvements through models, data optimization, tooling, and other technical innovations
• Architect and implement core components of the search platform and model infrastructure
• Design, train, and optimize large-scale deep learning models using PyTorch, distributed training frameworks (PyTorch Distributed, DeepSpeed, FSDP), and hardware acceleration
• Conduct advanced research in representation learning, including contrastive learning, multilingual, and multimodal modeling for search applications
• Deploy models—ranging from boosting algorithms to large language models—in scalable, high-performance production environments
• Build and optimize retrieval-augmented generation (RAG) pipelines for grounding and answer generation
• Collaborate closely with Data, AI, Infrastructure, and Product teams to ensure rapid, high-quality delivery

What We're Looking For

• Deep understanding of search and retrieval systems, including quality evaluation principles and metrics
• Proven track record building and deploying large-scale search or recommender systems
• Strong proficiency with PyTorch, including hands-on experience with distributed training techniques and performance optimization for large models
• Expertise in representation learning, including contrastive learning and embedding space alignment for multilingual and multimodal applications
• Strong publication record at top-tier AI/ML conferences or workshops (NeurIPS, ICML, ICLR, ACL, CVPR, SIGIR)
• Self-driven with a strong sense of ownership and execution
• Minimum 3 years of experience working on search, recommender systems, or closely related research areas (5+ years preferred)

How We Work with You

Candidates apply directly through Get A Job.ai. Our talent team will conduct an initial screening to understand your background and fit for the role. Once we confirm mutual interest, we submit your profile to our client for consideration. Please do not attempt to contact the client directly, as all communication is managed through our recruiting process.

Pay

Compensation details will be discussed during the screening process based on experience and qualifications.

Equal Opportunity

Get A Job.ai is committed to equal employment opportunity regardless of race, color, religion, sex, national origin, age, disability, or any other protected characteristic.