Associate, AI Engineer - Pune - Multiple roles

Overview

Early-career Full Stack Engineer focused on AI Engineering, building intelligent applications powered by large language models (LLMs), with RAG pipelines, LLM evaluation and agent-based architectures on Azure and Databricks.

What you'll do

• Build end-to-end AI-powered applications (UI, backend services and model orchestration)
• Design and implement RAG pipelines for enterprise knowledge retrieval and grounding
• Develop and deploy LLM-based workflows and agents (multi-step reasoning, tool use, orchestration)
• Integrate models via APIs and hosted platforms (Databricks, Azure AI Foundry, etc.)
• Develop LLM evaluation frameworks (accuracy, hallucination detection, relevance, safety)
• Implement automated testing and benchmarking for prompt chains and agents
• Define and monitor LLM performance metrics (precision, recall, grounding fidelity)
• Support experimentation (prompt engineering, model comparisons, fine-tuning readiness)
• Build and optimize data pipelines leveraging Databricks (Spark, Delta)
• Design and query structured and semi-structured datasets using Postgres
• Enable efficient retrieval layers (vector search, embeddings, indexing strategies)
• Support data modeling for AI use cases (knowledge bases, feature stores)
• Develop AI agents within Databricks and Azure ecosystems
• Integrate orchestration frameworks (LangChain, LangGraph, AutoGen, or similar)
• Build reusable services/APIs for agent capabilities (search, summarization, reasoning)
• Contribute to lightweight front-end interfaces for AI applications (React or similar)
• Build backend APIs/services (Python-based frameworks)
• Ensure production readiness (logging, monitoring, scalability, security)

What you'll need

• 4+ years of relevant experience in AI engineering, building AI applications, LLM, Python
• B.E/B.Tech or above degree required

Details

• Location: Pune, India