Python AI Engineer - Assistant Vice President

Citi is seeking a Python and Generative AI Engineer to help design, build, and deploy advanced AI models and applications, including sophisticated chat experiences. This onsite role in Irving, TX focuses on delivering performant, scalable solutions across the full AI lifecycle, from research and prototyping to production deployment with responsible and ethical practices.

Key Responsibilities

• Design, develop, and implement Generative AI models such as LLMs, GANs, VAEs, and Diffusion Models using Python and deep learning frameworks including TensorFlow and PyTorch.
• Integrate Generative AI capabilities into existing systems and develop new AI-powered applications.
• Build and maintain dynamic, responsive Generative AI chat interfaces using React, including HTTP streaming for real-time updates, asynchronous processing, secure file upload capabilities, and advanced chat formatting.
• Research and remain current on advancements in Generative AI, natural language processing, and machine learning.
• Optimize and fine-tune models for performance, scalability, and efficiency.
• Develop robust data pipelines for training, evaluation, and deployment of AI models.
• Partner with data scientists, machine learning engineers, and product managers to define requirements and deliver solutions.
• Support responsible and ethical deployment of AI models in line with best practices and regulatory guidelines.
• Write clean, maintainable, well-documented code following software development best practices.
• Participate in code reviews, contribute to architecture discussions, and provide technical guidance.
• Troubleshoot and debug AI models and applications, identifying performance bottlenecks and resolving issues.

Required Qualifications

• 5-8 years of professional software development experience with a strong focus on Python.
• Proven experience with Generative AI models, including hands-on work with large language models (LLMs) or other generative architectures.
• Experience with Google ADK or similar Google AI/ML platforms and SDKs.
• Understanding of deep learning frameworks including TensorFlow, PyTorch, or JAX.
• Strong front-end development skills, particularly React, for building interactive user interfaces.
• Demonstrable experience implementing chat functionality including HTTP streaming, asynchronous operations, file upload handling, and chat formatting techniques.
• Solid understanding of machine learning principles, algorithms, and data structures.
• Experience with cloud platforms (AWS, Azure, or GCP) and MLOps practices for deploying and managing AI models.
• Familiarity with containerization technologies such as Docker and Kubernetes.
• Strong problem-solving skills, with the ability to work independently and collaboratively.
• Excellent communication and interpersonal skills.

Technologies

• Python, TensorFlow, PyTorch, JAX
• React, HTTP streaming
• Google ADK
• AWS, Azure, GCP, MLOps
• Docker, Kubernetes
• LLMs, GANs, VAEs, Diffusion Models

Salary and Location

• Location: Irving, TX (onsite)
• Salary range: USD 107,120 - 160,680 per year
• Time type: Full time
• Primary location: Irving Texas United States

Education

• Bachelor’s or Master’s degree in Computer Science, Artificial Intelligence, Machine Learning, or a related quantitative field

Benefits

• Medical, dental & vision coverage
• 401(k)
• Life, accident, and disability insurance
• Wellness programs
• Paid time off packages, including planned time off (vacation), unplanned time off (sick leave), and paid holidays
• Discretionary and formulaic incentive and retention awards (for eligible employees)

Anticipated posting close date: Oct 13, 2026.

Citi uses automated processing, including artificial intelligence, for legitimate business interests to identify and align a candidate’s skills and abilities with a specific job opening. Hiring processes and decisions, including determining suitability for a role, are conducted, checked, and decided by individuals, and automated processing does not involve relying on automatic or autonomous decision-making.