Role: AI/ML Engineer

Function: AI/ML

Location: Bangalore

Compensation: Competitive compensation

Industry: AI/ML, Consumer Technology

About Company

The company is building the AI layer for Bharat at India-scale. Backed by partnerships with global tech leaders like Meta and Google, the team is creating AI that serves the entire Indian user base—across languages, contexts, and daily needs. This is AI designed for real adoption, not experiments.

They bring a rare combination of deep India-first AI capability and unmatched India-scale distribution. The focus is a platform-and-product stack that makes AI useful, reliable, and safe for everyday consumers. It's engineered from day one for massive scale—100M+ users early and 1B-ready constraints on latency, cost, reliability, and safety.

Position Overview

You'll architect the core intelligence platform and agent orchestration layer that powers India's next-generation consumer AI experience. You'll build agentic pipelines, vector retrieval systems, and model inference infrastructure serving 100M+ users with real-world impact. This role offers direct ownership of AI systems designed for massive scale with fast iteration loops and tangible outcomes.

Role & Responsibilities

• Build and optimize agentic pipelines using LangChain, LangGraph, and MCP/tool-use frameworks for multi-step AI workflows
• Design and implement vector retrieval systems with Pinecone, Milvus, and FAISS for semantic search applications
• Develop robust model inference pipelines with evaluation frameworks for quality, latency, and cost optimization
• Create embedding workflows, prompt templates, and structured output validation systems for reliable AI responses
• Integrate AI components into scalable platform APIs using Python and FastAPI for production deployment
• Implement model routing and orchestration systems across multiple AI models and modalities
• Build evaluation harnesses for measuring AI system performance, safety, and reliability at scale

Must Have Criteria

• 5-10 years of experience in ML engineering or platform-oriented AI development
• Strong proficiency in Python programming and PyTorch framework
• Hands-on experience building RAG (Retrieval-Augmented Generation) systems and working with embeddings
• Practical experience with Large Language Models (LLMs) integration and optimization
• Experience working with vector databases including Pinecone, Milvus, or FAISS for semantic search
• Proven experience building AI systems serving 100M+ users or processing billions of records
• Experience building APIs and backend services for AI/ML applications

Nice to Have

• Experience with Kubernetes, AWS, or Azure for model deployment and orchestration
• Background in building multi-agent systems or agentic AI workflows
• Experience with model evaluation frameworks and AI safety implementations
• Previous work at AI-first companies or building consumer-scale AI products

What We Offer

• Opportunity to build AI systems serving 100M+ users with 1B-ready scale constraints
• Work with cutting-edge AI technologies in a fast-paced, high-ownership environment
• Direct impact on India's AI ecosystem through partnerships with global tech leaders
• Competitive compensation and equity in a high-growth AI company
• Collaborative culture focused on engineering excellence and tangible impact