Gen AI/Python Engineer – Assistant Vice President

Citi; Pune District; On-site; INR 1,800,000 - 3,000,000; Full time

Job summary

Citi in Pune, India, is seeking a Senior AI/ML Engineer to lead generative AI initiatives within the Systems & Engineering group. You will work with PyTorch, TensorFlow, and Keras, and apply prompt engineering, RAG techniques, and model deployment in Dockerized environments.

You’ll collaborate with cross-functional teams, maintain code in Git, and ensure responsible AI practices. A 7+ year track record in AI/ML, strong Python skills, and cloud platform exposure are required or highly desired.

Qualifications

• Bachelor’s or Master’s in Computer Science, Data Science, AI, or related field.
• 7+ years of AI/ML development experience focusing on generative AI.
• Experience delivering AI/ML projects collaboratively.
• Exposure to cloud AI/ML environments (AWS/GCP/Azure) is a plus.

Skills

PyTorch

TensorFlow

Keras

LLM Prompt Engineering

RAG Systems

Hybrid Search

Docker

Python

LangChain

Hugging Face

LlamaIndex

Git

Responsible AI

Education

Bachelor’s or Master’s in Computer Science, Data Science, AI, or related field

7+ years of AI/ML development experience

Exposure to cloud AI/ML environments (AWS, GCP, Azure)

Tools

Docker

Git

LangChain

Hugging Face

LlamaIndex

Job description

Job Family Group: Technology

Job Family: Systems & Engineering

Time Type: Full time

• Major machine learning framework (PyTorch, TensorFlow, Keras).
• Familiarity with LLMs, prompt engineering, and fine-tuning (e.g., LoRA, QLoRA).
• Exposure to RAG systems and basic knowledge of hybrid search techniques.
• Understanding of model deployment and containerization (Docker).
• Proficiency in Python for AI development, data preprocessing, and scripting.
• Experience with generative AI tools (LangChain, Hugging Face, LlamaIndex) is a plus.
• Understanding of version control systems (Git).
• Awareness of AI compliance, data privacy, and responsible AI principles.

Required Soft Skills:

• Strong teamwork and communication abilities.
• Willingness to learn new AI/ML technologies and frameworks.
• Analytical mindset and attention to detail.
• Openness to feedback and continuous improvement.

Qualifications:

• Bachelor’s degree or Master’s degree in Computer Science, Data Science, AI, or a related field.
• 7 years of professional experience in AI/ML development, with exposure to Generative AI.
• Experience delivering AI/ML projects in a collaborative setting.
• Exposure to cloud-based AI/ML environments (AWS, GCP, or Azure) is a plus.

Citi is an equal opportunity employer, and qualified candidates will receive consideration without regard to their race, color, religion, sex, sexual orientation, gender identity, national origin, disability, status as a protected veteran, or any other characteristic protected by law. If you are a person with a disability and need a reasonable accommodation to use our search tools and/or apply for a career opportunity review Accessibility at Citi. View Citi’s EEO Policy Statement and the Know Your Rights poster.