Senior ML Engineer (Agentic Search)

Nebius Group; United States; Remote; USD 150,000 - 230,000; Full time

Job summary

Nebius is seeking a Senior ML Engineer to design, train, and deploy ML models for retrieval, ranking, and indexing at scale. You will build embedding‑based indexing and large‑scale retrieval systems, and develop models for crawling, data selection, and content understanding.

You will join a fast-growing team building an agent-native search platform powering AI systems and real-time information access. Expect production-grade challenges and close collaboration with engineers.

Qualifications

• 5+ years of software engineering or machine learning experience.
• Proficient in Python, Go, or C++.
• Experience deploying ML models in production.
• Experience with retrieval, ranking, or similar ML problems.
• Knowledge of ML and modern deep learning techniques.
• Experience with large-scale data systems and high-throughput environments.
• Ability to design evaluation frameworks and define meaningful model metrics.
• Product-oriented mindset with a focus on impact and iteration.
• Ability to work in a distributed team.

Responsibilities

• Design, train, and deploy ML models for retrieval, reranking, and search relevance in production
• Build embedding-based indexing and large-scale retrieval systems
• Develop models supporting crawling, data selection, and content understanding
• Define and improve quality metrics for agent-native search and build evaluation pipelines
• Work on systems operating at very large scale, including high-throughput query workloads
• Collaborate closely with engineering teams to integrate ML models into production services
• Analyse performance trade-offs across latency, quality, and cost
• Experiment with and apply state-of-the-art techniques in search, retrieval, and LLM-integrated systems
• Contribute to product and architectural decisions in a fast-moving environment

Skills

ML model design

Evaluation metrics

Large-scale data processing

Creative problem solving

Distributed teamwork

Tools

Python

Go

C++

Job description

About Nebius:

Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to production deployment, without the cost and complexity of building large in-house AI/ML infrastructure.Built by engineers, for engineers. From large-scale GPU orchestration to inference optimization, we own the hard problems across compute, storage, networking and applied AI.Listed on Nasdaq (NBIS) and headquartered in Amsterdam, we have a global footprint with R D hubs across Europe, the UK, North America and Israel. Our team of 1,500+ includes hundreds of engineers with deep expertise across hardware, software and AI R D.We are seeking a Senior ML Engineer to join a fast-growing team building an agent-native search platform for AI systems, the emerging web access layer for AI.You will develop and deploy machine learning models that power retrieval, ranking, and indexing at scale, helping AI systems access fresh, reliable information in real time.This is a high-impact role working on a production system used 24x7, tackling challenges comparable to large-scale web search.

Your responsibilities:

• Design, train, and deploy ML models for retrieval, reranking, and search relevance in production
• Build and optimise embedding-based indexing and large-scale retrieval systems
• Develop models supporting crawling, data selection, and content understanding
• Define and improve quality metrics for agent-native search and build evaluation pipelines
• Work on systems operating at very large scale, including high-throughput query workloads
• Collaborate closely with engineering teams to integrate ML models into production services
• Analyse performance trade-offs across latency, quality, and cost
• Experiment with and apply state-of-the-art techniques in search, retrieval, and LLM-integrated systems
• Contribute to product and architectural decisions in a fast-moving environment

Must-haves:

• 5+ years of experience in software engineering or applied machine learning
• Strong programming skills in Python, Go, or C++
• Proven experience deploying ML models in production systems
• Hands-on experience with retrieval, ranking, recommendation, or similar ML problems
• Strong understanding of machine learning and modern deep learning techniques
• Experience working with large-scale data systems and high-throughput environments
• Ability to design evaluation frameworks and define meaningful model metrics
• Product-oriented mindset with a focus on impact and iteration
• Strong problem-solving skills and ability to work in a distributed team

Nice-to-haves:

• Experience with search systems or large-scale information retrieval
• Familiarity with embeddings, transformers, and modern NLP systems
• Experience working on LLM-powered or agent-based systems
• Contributions to open-source projects, technical publications, or conference talks
• Participation in competitive ML (e.g. Kaggle) or simi