Python AI/ML Development Engineer

Discover your future at Citi

Working at Citi is far more than just a job. A career with us means joining a team of more than 230,000 dedicated people from around the globe. At Citi, you’ll have the opportunity to grow your career, give back to your community and make a real impact.

Job Overview

Python AI/ML Development Engineer

About the Role

Citi is seeking a highly skilled and experienced Python AI/ML Platform Engineer to join our innovative team. In this pivotal role, you will be instrumental in designing, developing, and implementing scalable and robust Python-based platforms that power our next-generation AI-driven applications and systems. You will work with cutting-edge Generative AI models and architectures, contributing to the strategic advancement of AI capabilities within the financial industry. This is an exciting opportunity for a passionate engineer to drive technological innovation and deliver high-impact solutions in a dynamic and challenging environment.

Key Responsibilities

• Design, develop, and maintain robust, scalable, and high-performance Python-based platforms for AI-driven applications and systems.
• Implement backend services and APIs, ensuring seamless integration with various internal and external systems.
• Leverage salient Python libraries and frameworks such as Fast API, Flask, Django, Pandas, NumPy, Prefect, Scikit-learn, and Hugging Face to build sophisticated AI/ML solutions.
• Apply deep expertise in Generative AI models and architectures, including transformers, large language models (LLMs like Llama 3, Llama 4, Gemini, GPT-4), Generative Adversarial Networks (GANs), and diffusion models.
• Incorporate solid understanding and principles of Agentic AI into platform design and implementation.
• Contribute to the strategic evolution of our cloud architecture, with a focus on public cloud offerings and best practices.
• Collaborate with AI/ML researchers, data scientists, and product teams to translate complex requirements into technical designs and scalable solutions.
• Ensure the delivery of production-grade solutions, adhering to best practices in software development, including agile methodologies, CI/CD pipelines, comprehensive software testing, and rigorous code reviews.
• Actively participate in code reviews, providing constructive feedback and ensuring high code quality and maintainability.
• Continuously research and integrate new AI/ML technologies and tools into our platforms to enhance capabilities and performance.
• Effectively communicate technical concepts and solutions to diverse audiences, both technical and non-technical.
• Mentor junior engineers and contribute to a culture of continuous learning and technical excellence.

Qualifications

• 6+ yearsof relevant experience in developing and implementing scalable and robust Python-based platforms for AI-driven applications and systems.
• 5+ yearsof experience delivering complex Python-based solutions and services (e.g., backend services, APIs) into production environments.
• Proven hands-on experience utilizing salient Python libraries such as Fast API, Flask, Django, Pandas, NumPy, Prefect, Scikit-learn, and Hugging Face.
• Deep understanding of Generative AI models and architectures, including transformers, LLMs (Llama 3, Llama 4, Gemini, GPT-4), GANs, and diffusion models. Solid understanding of Agentic AI principles is essential.
• Experience with cloud architectures, with specific experience in public cloud offerings (e.g., AWS, Azure, GCP).
• Strong passion and proven hands-on experience integrating with AI/ML technologies, particularly using Python.
• Proficiency in modern software development practices, including agile development, CI/CD pipelines, comprehensive software testing, and peer code reviews.
• Ability to quickly learn new technologies, adjust priorities in a fast-paced environment, influence highly skilled engineering teams, and guide technology decisions.
• Excellent communication and documentation skills, with the ability to communicate clearly and concisely, both written and verbal.
• Strong analytical, problem-solving skills, and a strategic mindset.
• Experience in the financial industry is a significant advantage.

Education

• Bachelor’s or Master’s degree in Computer Science, Engineering, Artificial Intelligence, or a related technical field.

------------------------------------------------------

Job Family Group:

Technology

------------------------------------------------------

Job Family:

Applications Development

------------------------------------------------------

Time Type:

Full time

------------------------------------------------------

Most Relevant Skills

Please see the requirements listed above.

------------------------------------------------------

Other Relevant Skills

For complementary skills, please see above and/or contact the recruiter.

------------------------------------------------------

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.