Python - Gen AI Developer - Assistant Vice President

Citi; Maharashtra; On-site; INR 1,800,000 - 3,200,000; Full time

Job summary

Citi is seeking a Python/Generative AI Developer to design, develop, and deploy advanced AI models and applications, including chat interfaces. The role spans research to production, collaborating with data scientists and product teams to deliver scalable solutions.

Candidates should have 8+ years in software development with Python, strong Generative AI experience, and proficiency in TensorFlow, PyTorch, or JAX, plus React front-end skills for interactive interfaces and chat functionality.

Qualifications

• Bachelor's or Master's degree in Computer Science, AI, ML, or related field.
• 8+ years of professional software development experience with a strong Python focus.
• Proven experience with Generative AI models (LLMs or other architectures).
• Proficiency with TensorFlow, PyTorch, or JAX.
• Experience with Google AI/ML platforms/SDKs.
• Strong React front-end development skills for interactive UIs.
• Experience implementing chat features with HTTP streaming and async processing.
• Knowledge of ML/DL concepts, data structures, and cloud/MLOps practices.
• Familiarity with Docker and Kubernetes for containerized deployments.

Responsibilities

• Design, develop, and deploy Generative AI models and applications.
• Build and maintain chat interfaces using React with real-time updates.
• Create data pipelines for training, evaluation, and deployment of AI models.
• Collaborate with data scientists and PMs to define requirements and deliver solutions.
• Ensure ethical and responsible deployment of AI models aligned with guidelines.
• Write clean, well-documented code and participate in reviews and architecture discussions.
• Troubleshoot performance bottlenecks and optimize models and pipelines.

Skills

Python

Generative AI

React

Communication

Education

Bachelor's or Master's in CS/AI or related

Tools

TensorFlow

PyTorch

JAX

Google AI/ML SDKs

Docker

Kubernetes

Job description

We are seeking a highly skilled and motivated Python/Generative AI Developer to join our growing team. The ideal candidate will be responsible for designing, developing, and deploying advanced Generative AI models and applications, including sophisticated chat interfaces. You will work on various aspects of the AI lifecycle, from research and prototyping to production deployment, contributing to solutions that drive innovation and efficiency within our organization.

Responsibilities

• Design, develop, and implement Generative AI models (e.g., LLMs, GANs, VAEs, Diffusion Models) using Python and relevant deep learning frameworks (e.g., TensorFlow, PyTorch).
• Integrate Generative AI capabilities into existing systems and develop new AI‑powered applications.
• Build and maintain dynamic and responsive chat interfaces for Generative AI solutions using React, incorporating features such as HTTP streaming for real‑time updates, asynchronous processing for efficient user experience, secure file upload capabilities, and advanced chat formatting.
• Conduct research and stay up‑to‑date with the latest advancements in Generative AI, natural language processing, and machine learning.
• Optimize and fine‑tune AI models for performance, scalability, and efficiency.
• Develop robust data pipelines for training, evaluation, and deployment of AI models.
• Collaborate with data scientists, machine learning engineers, and product managers to define requirements and deliver solutions.
• Ensure the responsible and ethical deployment of AI models, adhering to best practices and regulatory guidelines.
• Write clean, maintainable, and well‑documented code, adhering to software development best practices.
• Participate in code reviews, contribute to architectural discussions, and provide technical guidance.
• Troubleshoot and debug AI models and applications, identifying and resolving performance bottlenecks and issues.

Required Qualifications

• Bachelor's or Master's degree in Computer Science, Artificial Intelligence, Machine Learning, or a related quantitative field.
• 8+ years of professional experience in software development, with a strong focus on Python.
• Proven experience with Generative AI models, including hands‑on experience with large language models (LLMs) or other generative architectures.
• Proficiency in deep learning frameworks such as TensorFlow, PyTorch, or JAX.
• Experience with Google ADK or similar Google AI/ML platforms and SDKs.
• Strong front‑end development skills, particularly with React, for building interactive user interfaces.
• Demonstrable experience in implementing chat functionalities, including handling HTTP streaming, asynchronous operations, managing file uploads, and applying various chat formatting techniques.
• Solid understanding of machine learning principles, algorithms, and data structures.
• Experience with cloud platforms (e.g., AWS, Azure, GCP) and MLOps practices for deploying and managing AI models.
• Familiarity with containerization technologies (e.g., Docker, Kubernetes).
• Strong problem‑solving skills and the ability to work independently and as part of a team.
• Excellent communication and interpersonal skills.

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.