Pune · Hybrid

AI Engineer

Full-timePosted 3 Nov

Salary: ₹18.8–20.7 LPAEstimated from AmbitionBox
Experience: 3–4 years
Eligible batches: 2022 · 2023
Location: PuneHybrid

About the role

The AI Engineer will design and deliver generative‑AI, multimodal and agentic solutions for global brands. Working from Pune, the candidate will turn transformer‑based models into production‑grade APIs, integrate knowledge‑graph retrieval, and collaborate with data, CX and engineering teams.

Responsibilities

Build conversational, multimodal and agentic AI applications using LLMs and frameworks such as LangChain, LangGraph or LlamaIndex.
Design AI workflows that incorporate reasoning, planning, tool‑use, memory and external system integration.
Develop knowledge‑graph‑enhanced retrieval pipelines and ensure safety through evaluation and guardrails.
Deploy models as scalable APIs or micro‑services with Python, FastAPI/Flask, Docker and cloud platforms (AWS, Azure, GCP).
Collaborate with MLOps teams on CI/CD, model versioning, monitoring and cost‑optimized cloud operations.
Mentor junior engineers, document experiments and contribute reusable components.

Free interview prep

Prepare for the interview

All lists

Picked for this role's title and skills: the questions that come up most, each with a short answer to say out loud.

Machine learning interview questions40 questions · ≈ 18 min to revise

Asked most often · 19 of 40

Eligibility

Minimum 3‑4 years of software development experience with production ML deployments.
At least 2 years of hands‑on work with deep‑learning, generative AI or transformer architectures.
Strong problem‑solving ability and clear communication with non‑technical stakeholders.

Skills

Advanced Python programming and SQL.API development with Flask, FastAPI or Django.Deep‑learning frameworks: PyTorch, TensorFlow.Cloud services: AWS, Azure, GCP.Big‑data tools: Apache Spark, Hadoop, MongoDB.