ML/AI Engineer

Endee; Bengaluru; On-site; INR 1,200,000 - 1,800,000; Full time

Job summary

A leading tech company in Bengaluru is seeking an experienced ML/AI Engineer to architect the intelligence layer of their vector database. The candidate will design embedding pipelines and implement advanced search algorithms while collaborating with the database team. Solid experience in ML/AI and proficiency in Python are crucial. The role impacts how developers use AI applications, making it vital for innovative product features.

Qualifications

• 3+ years of hands-on experience in ML/AI with production deployment.
• Deep proficiency in Python.
• Expertise with embeddings, vector search, or recommendation systems.

Responsibilities

• Research and prototype cutting-edge vector search algorithms.
• Design and optimize end-to-end embedding pipelines.
• Develop retrieval-augmented generation integrations.

Skills

ML/AI experience

Python proficiency

C/C++ experience

Understanding of transformer architectures

Experience with PyTorch

Tools

PyTorch

TensorFlow

JAX

Job description

Join Endee as an ML/AI Engineer to architect and build the intelligence layer of our vector database. You will design embedding pipelines, implement state-of-the-art similarity search algorithms, and develop AI-powered features that serve millions of queries per second. This role sits at the intersection of applied research and production engineering — you will ship models that directly impact how developers build AI applications.

Responsibilities

• Research, prototype, and productionize cutting-edge vector search and approximate nearest neighbor (ANN) algorithms
• Design and optimize end-to-end embedding pipelines for low-latency, high-throughput inference
• Build evaluation frameworks and automated benchmarking suites for search quality and retrieval accuracy
• Develop retrieval-augmented generation (RAG) integrations and semantic search capabilities
• Collaborate with the core database team to co-design indexing structures optimized for ML workloads
• Partner with customers and developer advocates to translate real-world AI use cases into product features

Requirements

• 3+ years of hands-on experience in ML/AI, with production deployment experience
• Deep proficiency in Python; experience with C/C++ for performance-critical components is a strong plus
• Expertise with embeddings, vector search, information retrieval, or recommendation systems
• Solid understanding of transformer architectures, attention mechanisms, and modern LLM pipelines
• Experience with PyTorch, TensorFlow, or JAX in production settings
• Strong fundamentals in linear algebra, probability, and algorithm design