Sign up to save this job, get alerts, and apply with an optimized CV.

Log in Sign up free

Senior ML Engineer (Immediate Joiner)

Job description

About the Role

This role is for a hands-on ML Engineer who can design, train, and productionize models powering search relevance, retrieval, personalization, and LLM-based conversational experiences at a massive scale.

You will work closely with backend, platform, and catalog enrichment teams to deliver high-quality ML components under tight performance and latency constraints.

Key Responsibilities
  • Build and improve search ranking, retrieval, and query understanding models.
  • Develop ML components for Conversational Search:
  • Multi-turn context handling
  • Query intent detection and classification
  • Retrieval-augmented generation (RAG) pipelines
  • Reasoning workflows (ReAct, static + dynamic agent flows)
Design and optimize embedding models, vector stores, and similarity search systems.Build personalized ranking and recommendation models using deep learning.Work on large-scale ML systems optimized for:
  • Low latency
  • High throughput
  • Cost-efficient inference
Implement ML pipeline best practices (versioning, monitoring, A/B testing, observability).Collaborate with platform teams to integrate ML services across search, recommendations, and conversational agents.Develop caching strategies (prompt cache, vector cache, similarity caching) to hit strict SLA targets.Contribute to long-term roadmap: foundational retrieval models, multi-objective optimization, and user lifecycle modeling.



Requirements

Required Qualifications
  • 4-10 years of experience in Machine Learning / Applied ML engineering.
  • Strong foundations in ML, deep learning, Transformers, and neural retrieval.
  • Hands-on experience with:
  • Search systems (retrieval + ranking)
  • Recommendation models
  • Embedding models & vector databases
  • TensorFlow / PyTorch
Proven experience building production-grade ML systems at scale.Familiarity with LLMs, RAG architectures, prompt engineering, and agent workflows.Strong coding skills (Python) and experience with modern ML stack (TensorFlow, PyTorch, Faiss/ScaNN, Triton, etc.).Ability to work closely with backend teams to deploy models in distributed systems.Excellent problem-solving skills and comfort working on ambiguous, high-impact problems.

Preferred Qualifications
  • Experience with conversational AI, chat-based retrieval, or multi-turn dialog modeling.
  • Experience in media, streaming, sports data, or large catalog discovery.
  • Knowledge of micro-drama, short-video personalization, or multi-objective recommendation systems.
  • Strong understanding of scalability patterns: batching, async orchestration, and caching layers.

Benefits

What you get

    • Best in class salary: We hire only the best, and we pay accordingly.
    • Proximity Talks: Meet other designers, engineers, and product geeks — and learn from experts in the field.
    • Keep on learning with a world-class team: Work with the best in the field, challenge yourself constantly, and learn something new every day.

    About us

    Proximity is the trusted technology, design, and consulting partner for some of the biggest Sports, Media, and Entertainment companies in the world! We’re headquartered in San Francisco and have offices in Palo Alto, Dubai, Mumbai, and Bangalore. Since 2019, Proximity has created and grown high-impact, scalable products used by 370 million daily users, with a total net worth of $45.7 billion among our client companies.

    Today, we are a global team of coders, designers, product managers, geeks, and experts. We solve complex problems and build cutting-edge tech, at scale. Our team of Proxonauts is growing quickly, which means your impact on the company’s success will be huge. You’ll have the chance to work with experienced leaders who have built and led multiple tech, product, and design teams. Here’s a quick guide to getting to know us better:

    Here’s a quick glimpse of Proximity and what it’s like to be a Proxonaut:

    • Visit this YouTube link to listen to what our CEO, Hardik Jagda, has to say about Proximity.
    • Meet some of our Proxonauts here: Know thy Proxonauts better
    • Here are some quick links to the Careers page, Blog, and Studio Proximity (our design wing).

    Follow our team's #BTS (behind-the-scenes) updates on our Instagram channels —
    - @ProxWrks - @H.Jagda

    Required skills

    pytorch prompt engineering machine learning conversational ai tensorflow llms media rag architectures streaming sports data batching applied ml engineering search systems recommendation models embedding models & vector databases faiss/scann triton agent workflows chat-based retrieval multi-turn dialog modeling large catalog discovery micro-drama short-video personalization multi-objective recommendation systems scalability patterns async orchestration caching layers

    Sign up to apply

    Create a free account to apply for this job and get access to:

    • AI-powered CV optimization for this specific job
    • Save jobs and create custom alerts
    • See your CV match score for each job
    Create free account Log in

    Company information

    Company
    Proximity Works
    Location
    India
    India
    Posted
    8 months ago
    View all jobs at Proximity Works

    Find similar jobs

    Explore more opportunities like this one.

    Jobs in India

    Interested in this position?

    Create your free account and tailor your CV to match this job.

    Create free account Log in