AI / ML engineer

JD : AI / ML Python Engineer – RAG & LLM Systems

Location : Pune (On-Site)

Experience : 4+ years

Job Description

We are looking for a highly skilled AI / ML Python Engineer with hands-on experience in LLMs (Llama), RAG pipelines, vector databases, and backend API development. The ideal candidate is strong in Python, machine learning workflows, embeddings, and scalable backend architecture, with the ability to build reliable AI-driven applications end-to-end.

Key Responsibilities

AI / ML Development & Optimization

• Build and integrate GenAI and classical ML models using LangChain, LlamaIndex, Scikit-learn, TensorFlow, PyTorch .
• Optimize models via hyperparameter tuning, cross-validation , and monitor performance using observability tools.
• Improve accuracy, latency, and reliability of LLM-based pipelines.

RAG, Embeddings & Vector Search

Multimodal Data Processing

Backend & API Engineering

Database & Data Workflow Management

Task Automation & Scheduling

Containerization & Deployment

End-to-End Development

Required Skills

Nice to Have