Machine Learning Engineer Jobs in Pune – AI, NLP, Python, FastAPI

DigitalXNode; Pune District; On-site; INR 1,500,000 - 3,000,000; Full time

Job summary

DigitalXNode in Pune is hiring AI/ML professionals to build scalable machine learning and generative AI solutions on production-grade systems using LLMs, FastAPI, Python, and NLP technologies. You'll design and deploy GenAI apps, implement RAG pipelines, develop AI agents with LangGraph, and work on knowledge graphs, APIs, and scalable architectures, with CI/CD and code reviews shaping the workflow.

3–12 years of experience welcomed, with a bachelor’s in CS or related field.

Qualifications

• Strong background in AI, ML, and GenAI.
• Hands-on with LLMs, RAG pipelines, NLP.
• Experience with LangGraph / AI workflow orchestration.
• Proficiency in Python, FastAPI, API development.

Responsibilities

• Design and deploy LLM-based and Generative AI applications.
• Build scalable RAG (Retrieval-Augmented Generation) pipelines.
• Develop AI agents and agentic workflows using LangGraph.
• Implement NLP solutions.
• Work with knowledge graphs and semantic layers.
• Build APIs using Python and FastAPI.
• Optimize SQL/NoSQL databases for performance.
• Contribute to system architecture, microservices, and scalable systems.
• Follow CI/CD, Git workflows, and code review practices.

Skills

AI/ML expertise

GenAI & NLP

LLMs & RAG pipelines

Python & FastAPI

SQL/NoSQL databases

System design & microservices

Education

Bachelor's degree in CS/related field

Tools

LangGraph

FastAPI

Python

APIs development

SQL/NoSQL databases

Job description

Exciting opportunity for AI/ML professionals in Pune to build scalable Machine Learning and Generative AI solutions.
Work on production-grade systems leveraging LLMs, FastAPI, Python, and NLP technologies.

Experience:

3-10 Years (flexible for exceptional candidates up to 12 years)

Key Responsibilities

• Design and deploy LLM-based and Generative AI applications
• Build scalable RAG (Retrieval-Augmented Generation) pipelines
• Develop AI agents and agentic workflows using LangGraph
• Implement NLP (Natural Language Processing) solutions
• Work with knowledge graphs and semantic layers
• Build APIs using Python and FastAPI
• Optimize SQL/NoSQL databases for performance
• Contribute to system architecture, microservices, and scalable systems
• Follow CI/CD, Git workflows, and code review practices

Required Skills

• Strong experience in Artificial Intelligence, Machine Learning (ML), and Generative AI (GenAI)
• Hands-on with LLMs, RAG pipelines, NLP
• Experience with LangGraph / AI workflow orchestration
• Proficiency in Python, FastAPI, API development
• Strong knowledge of SQL and NoSQL databases
• Experience in system design and scalable architecture

Good to Have

• Experience with vector databases (Pinecone, FAISS, Weaviate)
• Knowledge of cloud platforms (AWS, Azure, GCP)
• Exposure to MLOps and AI deployment pipelines

Education

Bachelor’s degree in Computer Science, Engineering, or related field (Any Graduate with relevant experience is welcome)