Senior AI Engineer, Python — Enterprise Risk Tech

Citigroup Inc.; Mississauga; On-site; CAD 130,000 - 196,000; Full time

Benefits offered by this job

Health insurance

Job summary

Citigroup Inc. in Mississauga is seeking a Python AI Engineer to join the Retail and Wealth Risk Engineering team within the Enterprise Risk Technology platform.

You will design, build, and maintain scalable data pipelines and AI-enabled solutions leveraging PySpark, Databricks, Kafka, and FAST API in a cloud-native environment. The role emphasizes ELT/ETL, data governance, and alignment with regulatory and risk controls while adopting AI-based use cases and modern data architectures such as

Qualifications

• 5-8 years of overall experience in large-scale application development with recent mandatory platform for the secure and scalable deployment of AI agents into application contexts.
• Minimum of 5+ years of proven experience in a Python and pyspark Engineering lead role focused on building enterprise-grade, high-volume ELT/ETL processes using the PySpark and Databricks ecosystem.
• Hands-on experience with agentic AI development using YAML, JSON, FAST API or Spring boot, Google ADK, LLM integrations, including Devin.AI or GitHub Copilot, and integrating models via platforms like MCP using advanced prompt engineering.
• Proven experience developing and automating microservice integrations to support data-intensive applications.
• Proficiency in at least one programming language commonly used for data analytics, engineering, such as Python or Scala.
• Strong SQL skills and experience with various relational databases.
• Deep understanding of data modeling, data warehousing concepts, Data Mesh architecture, and data federation.
• Excellent communication, collaboration, and problem-solving skills.

Responsibilities

• Design, develop, and maintain scalable, enterprise-grade AI agents, supporting ELT/ETL processes to handle large data volumes using the Python, FAST API, Microservices, PySpark, Kafka and Databricks ecosystem.
• Build and Deploy GEN AI Agents using Googles ADK and Google Flash 2.5+ LLMs to support application automation supports and its deep insights, workflow support with HIL - Human in loop architecture.
• Build and maintain data federation layers for lambda and Data Mesh architectures using tools like Starburst, with a strategy for adopting AI-based use cases to drive efficiency.
• Develop, deploy, and automate microservice integrations to support data-intensive applications, ensuring scalability, resilience, and maintainability using cloud native infrastructure and OpenShift or Kubernetes architecture including CI/CD pipelines.
• Integrate and leverage agentic AI tools and platforms through advanced prompt engineering to enhance development and operational efficiency.
• Ensure data quality, integrity, and security throughout the entire data lifecycle.
• Contribute to the continuous improvement of data engineering processes, standards, and best practices within the team.
• Appropriately assess risk when business decisions are made, demonstrating consideration for Citi's reputation and safeguarding Citi, its clients, and assets by driving compliance with applicable laws, rules, and regulations.

Skills

Python

PySpark

Databricks

Kafka

FAST API

Microservices

SQL

Cloud

Kubernetes

AI/ML

Education

Bachelor's degree in Computer Science

Master's degree is a plus

Tools

Docker

GitHub Copilot

Google ADK

MCP

Job description

Citigroup Inc. in Mississauga is seeking a Python AI Engineer to join the Retail and Wealth Risk Engineering team within the Enterprise Risk Technology platform.

You will design, build, and maintain scalable data pipelines and AI-enabled solutions leveraging PySpark, Databricks, Kafka, and FAST API in a cloud-native environment. The role emphasizes ELT/ETL, data governance, and alignment with regulatory and risk controls while adopting AI-based use cases and modern data architectures such as