Senior Python AI Engineer – Generative AI Architect

Citigroup Inc.; Irving (TX); On-site; USD 107,000 - 161,000; Full time

Job summary

Citigroup Inc. seeks a Python/Generative AI Developer to design, develop, and deploy advanced AI models and chat interfaces.

You will work across AI lifecycle, from research to production, integrating AI capabilities into existing systems and collaborating with data scientists, engineers, and product managers. You will optimize models for performance and scalability, ensure responsible deployment, and contribute to robust data pipelines and secure interfaces.

Qualifications

• 5–8 years of software development experience with Python.
• Hands-on experience with Generative AI models and LLMs.
• Experience with Google ADK or similar AI/ML platforms and SDKs.
• Deep learning frameworks such as TensorFlow, PyTorch, or JAX.
• Strong React front-end skills for interactive UIs.
• Experience with cloud platforms (AWS/Azure/GCP) and MLOps.
• Familiarity with Docker and Kubernetes.

Responsibilities

• Design, develop, and deploy Generative AI models and apps.
• Build AI-powered chat interfaces with real-time updates.
• Create data pipelines for training and deployment.
• Collaborate with data scientists and product teams.
• Ensure responsible AI deployment and coding best practices.
• Troubleshoot performance bottlenecks in AI systems.

Skills

Python

Generative AI

LLMs

TensorFlow

PyTorch

JAX

React

HTTP streaming

Asynchronous processing

Cloud platforms

Docker

Kubernetes

Education

Bachelor's or Master's in Computer Science/AI/ML

Tools

Google ADK

TensorFlow

PyTorch

JAX

Docker

Kubernetes

Job description

Citigroup Inc. seeks a Python/Generative AI Developer to design, develop, and deploy advanced AI models and chat interfaces.

You will work across AI lifecycle, from research to production, integrating AI capabilities into existing systems and collaborating with data scientists, engineers, and product managers. You will optimize models for performance and scalability, ensure responsible deployment, and contribute to robust data pipelines and secure interfaces.