Staff Software Engineer, Machine Learning

We are looking for a Staff Machine Learning Engineer to play a pivotal role within the AI team. This role goes beyond building models - you will set the technical direction and lead initiatives that push the boundaries of what's possible. You will combine deep research expertise with the ability to build production-grade ML systems at scale, partnering across engineering, product, and research to deliver transformative AI solutions.

Responsibilities

• Own end-to-end model development and deployment of embeddings, retrieval, ranking, and classification models in large-scale production environments.  

• Build and deploy relevance models that power our various recommender systems and personalized intent engines.  

• Explore novel approaches for machine learning ranking and modeling downstream impact, as well as new ways of creating new product features by leveraging the cutting edge machine learning techniques in Generative AI, NLP.  

• Keep scalability and performance in mind through your design and engineering choices.  

• Work directly with partner data science, product, and infrastructure teams to take your ideas from conception to production.  

• Mentor junior ML engineers in applying advanced machine learning techniques to core problems such as entity resolution, lead scoring, and search relevance.  

Basic Qualifications

• MS/PhD degree in Computer Science or related technical discipline, or equivalent practical experience.

• 8+ years of experience in machine learning, artificial intelligence, or related fields.

• 5+ years in an architect or technical leadership role driving large-scale AI initiatives.

• Hands-on experience with end-to-end ML lifecycle: data pipelines, training, deployment, monitoring, and optimization.

• Expertise in at least one of: LLMs, generative AI, search/recommendations, reinforcement learning, or large-scale deep learning.

Preferred Qualifications

• 10+ years of experience building and scaling machine learning systems in production.

• Proven experience in pre-training and fine-tuning LLMs, including domain-specific adaptations.

• Strong software engineering skills (Python, Java, C++, or similar), with the ability to design robust, scalable systems.

• Track record of influencing product and technical strategy through AI/ML innovation.

• Publications, patents, or open-source contributions in machine learning, NLP, or related areas.

Suggested Skills

• Machine Learning

• Large Language Models

• Generative AI

• Search & Recommendations

• Reinforcement Learning

• Technical Leadership

• Cross-Functional Collaboration

You Will Benefit from Our Culture

We strongly believe in the well-being of our employees and their families. That is why we offer generous health and wellness programs and time away for employees of all levels.

India Disability Policy 

Global Data Privacy Notice for Job Candidates ​

India Disability Policy 

Global Data Privacy Notice and Compliance Posters for Job Candidates