Generative AI Engineer

BharatGen; Mumbai; On-site; INR 2,500,000 - 4,500,000; Full time

Job summary

BharatGen seeks a Generative AI Engineer to design, build, fine-tune, and deploy cutting-edge generative AI models and agentic systems. You will work on the full lifecycle of foundational model development from LLMs to real-world production impact.

Ideal candidates have 2–5 years of applied ML/AI experience, strong Python and ML framework skills, and experience with distributed training, MLOps, and open-source AI projects.

Qualifications

• Bachelor’s or Master’s in Computer Science, AI/ML, Data Science or related field, 2–5 years in applied ML/AI
• Proficiency in Python and CS fundamentals (data structures & algorithms)
• Experience with major ML frameworks (TensorFlow, PyTorch) and open-source AI tools
• Hands-on with model libraries and orchestration tools (Langchain, Langgraph, LlamaIndex)
• Experience with GPU/TPU training and inference, distributed training tools
• Familiarity with MLOps, version control, observability and data tooling (Git, DVC, MLFlow, W&B)
• Knowledge of data processing and vector databases (Dask, Milvus, Pinecone)
• Strong collaboration and ability to work with remote teams
• Track record of building/deploying LLMs/SLMs and contributing to open-source
• Startup mindset with proactive experimentation and problem-solving

Responsibilities

• Develop and train generative AI models across modalities (text, speech, vision)
• Fine-tune models for domain-specific tasks
• Build data pipelines for curation, training, evaluation, and improvement
• Implement CI/CD and observability for reliable development
• Develop RAG pipelines and optimize prompt strategies
• Optimize training/inference with quantization and distributed acceleration
• Monitor and improve model accuracy and efficiency, reduce hallucinations
• Collaborate with cross-functional teams to deploy AI stacks
• Document architectures, processes, and results with version control

Skills

Python programming

Problem solving

Project leadership

TensorFlow

PyTorch

HF Transformers

Langchain

LlamaIndex

vLLM

Distributed training

Git

DVC

MLFlow

Kubeflow

W&B

Spark/Dask

Pinecone

Milvus

Education

Bachelor’s or Master’s in CS/AI

Tools

Git

DVC

MLFlow

Kubeflow

W&B

SLURM

Ray

PyTorch

TensorFlow

HF Transformers

Langgraph

Langchain

LlamaIndex

Pinecone

Milvus

Job description

We are looking for a Generative AI – Engineer to join our growing team to help design, build, fine-tune, and deploy cutting – edge generative AI models and agentic systems. You will work on the full lifecycle of foundational model development – involving both large and small language models (LLMs and SLMs) – to create scalable AI solutions that address diverse needs across different business domains. This role is ideal for proactive individuals with a strong foundation in machine learning and an experimental mindset, who are passionate about driving transformative advancements in generative AI from research to real-world production impact.

Key Responsibilities

• Develop and train foundational generative AI models across modalities such as text-to-text, text-to-speech, automatic speech recognition, and vision language.
• Fine-tune and adapt models for specific tasks and domains.
• Build and maintain pipelines for data curation, preprocessing, training, evaluation, and continuous improvement of models.
• Implement debugging, CI/CD, and observability to ensure reliability and efficiency across the development lifecycle.
• Develop retrieval-augmented generation (RAG) pipelines and optimize prompt engineering strategies.
• Optimize training and inference performance through quantization, distributed training/inference, GPU/TPU acceleration.
• Monitor, benchmark, and improve model performance with a focus on accuracy, efficiency, and reducing hallucinations.
• Collaborate with cross-functional teams to build robust AI stacks and integrate them seamlessly into production pipelines for deployment.
• Document technical processes, AI model architectures, and experimental results, while maintaining well- structured, version-controlled code repositories.
• Stay current with advancements in transformer architectures, open-source releases, and AI tooling.

Minimum Qualifications and Experience

• Bachelor’s or Master’s in Computer Science, AI/ML, Data Science or any related field with 2 to 5 years of industry experience in applied machine learning or AI development.

Required Expertise

• Proficiency in Python programming with solid foundation in computer science fundamentals such as data structures and algorithms.
• Strong problem-solving skills and demonstrated ability to lead projects.
• Hands-on experience with a few of the tools listed below:
• One or more model libraries and ML frameworks such as TensorFlow, PyTorch, HF Transformers, NeMo, etc.
• AI application libraries and orchestration frameworks such as DSPy, Langgraph, Langchain, Llamaindex, etc.
• GPU/TPU based training and inference using libraries such as vLLM.
• Distributed training tools such as SLURM, Ray, Pytorch DDP, NCCL, etc.
• Version control, observability systems, and MLOps tools such as Git, DVC, W&B, MLFlow, KubeFlow, etc.
• Data analysis and curation tools such as Dask, Milvus, Apache Spark, Numpy, etc.
• Chunking, embeddings, vector databases (e.g., Pinecone, Weaviate, Milvus), and retrieval-augmented generation (RAG).
• Model context protocol (MCP), Agent to Agent (A2A), and Agent Communication Protocol (ACP).
• Team player with excellent interpersonal skills and ability to collaborate effectively with remote team members.
• Go-getter attitude and ability to flourish in a fast-paced, startup environment.
• Prior experience of building and deploying LLMs or SLMs, experience with multimodal models, and track record of contributions to open-source AI/ML projects would be a big plus.