Python Engineer - AI/ML

Posted 2 Days Ago
Be an Early Applicant
Pune, Mahārāshtra, IND
In-Office
Mid level
Fintech • Financial Services
The Role
Design, build, and deploy scalable AI agents and predictive analytics solutions using frameworks like LangChain and LlamaIndex. Integrate pre-trained LLMs, develop data ingestion pipelines, collaborate with stakeholders, perform testing and security analysis, and support production-grade AI applications in the financial services context.
Summary Generated by Built In

Seeking an experienced AI/ML Engineer to build sophisticated AI agents and predictive analytics tools. The role involves leveraging AI frameworks to create practical, high-impact solutions. You will focus on architecting intelligent systems that solve complex business challenges, moving beyond model creation. This is an opportunity for a skilled engineer to innovate at the forefront of applied artificial intelligence.
Responsibilities:

  • The ideal candidate will specialize in the design, development, and deployment of sophisticated AI-powered tools and autonomous agents. This role is focused on leveraging state-of-the-art machine learning models and frameworks to build practical, high-impact solutions, including predictive analytics platforms and intelligent automation systems. This is a pivotal role for an engineer who is passionate about applying cutting-edge AI to solve complex business challenges, moving beyond model creation to architecting intelligent, agent-based systems.
  • Architect, build, and deploy robust, scalable AI agents and tools using frameworks like LangChain, LlamaIndex, and others. Your focus will be on creating systems that can reason, act, and automate complex workflows.
  • Integrate pre-trained Large Language Models (LLMs) and other foundational models into cohesive, end-to-end applications. Develop and optimize data ingestion and processing pipelines for these systems.
  • Collaborate with product owners, data scientists, and business stakeholders to define requirements, scope projects, and deliver AI solutions that meet strategic objectives.
  • Consult with users, clients, and other technology groups on issues, and recommend programming solutions, install, and support customer exposure systems
  • Apply fundamental knowledge of programming languages for design specifications.
  • Analyze applications to identify vulnerabilities and security issues, as well as conduct testing and debugging
  • Serve as a subject matter expert in the practical application of various AI/ML frameworks. Evaluate, select, and optimize the right tools and techniques for specific use cases.
  • Resolve issues by identifying and selecting solutions through the applications of acquired technical experience and guided by precedents
  • Has the ability to operate with a limited level of direct supervision.
  • Can exercise independence of judgement and autonomy.
  • Appropriately assess risk when business decisions are made, demonstrating particular consideration for the firm's reputation and safeguarding Citigroup, its clients and assets, by driving compliance with applicable laws, rules and regulations, adhering to Policy, applying sound ethical judgment regarding personal behavior, conduct and business practices, and escalating, managing and reporting control issues with transparency.

Qualifications:

  • 2-6 years of relevant experience in the Financial Service industry
  • AI Frameworks: Hands-on, practical experience with modern AI development frameworks such as LangChain, LlamaIndex, or similar agent-based and orchestration tools.
  • Python Expertise: Advanced proficiency in Python and its ecosystem of libraries for data science and machine learning (e.g., Pandas, NumPy, Scikit-learn).
  • Consistently demonstrates clear and concise written and verbal communication
  • Exceptional problem-solving skills with the ability to design effective, scalable solutions for complex and often ambiguous problems.
  • Ability to work under pressure and manage deadlines or unexpected changes in expectations or requirements

Education:

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

This job description provides a high-level review of the types of work performed. Other job-related duties may be assigned as required.

  • Experience with vector databases and search technologies (e.g., Pinecone, Chroma, FAISS, Elasticsearch).
  • A portfolio of projects (e.g., on GitHub) demonstrating the construction of AI agents, tools, or complex application workflows.
  • Experience building and consuming APIs (REST, gRPC).
  • Familiarity with data engineering and building scalable data pipelines.

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

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.

Skills Required

  • 2-6 years of relevant experience in the Financial Service industry
  • Hands-on experience with AI frameworks such as LangChain or LlamaIndex
  • Advanced proficiency in Python and ML/data libraries (Pandas, NumPy, Scikit-learn)
  • Designing, building, and deploying AI agents and intelligent automation systems
  • Experience integrating pre-trained LLMs and foundational models into applications
  • Ability to analyze applications for vulnerabilities, perform testing and debugging
  • Clear and concise written and verbal communication skills
  • Bachelor's or Master's degree in Computer Science, Engineering, AI, or related field
  • Experience with vector databases and search technologies (Pinecone, Chroma, FAISS, Elasticsearch)
  • Portfolio of projects (e.g., GitHub) demonstrating AI agents or complex application workflows
  • Experience building and consuming APIs (REST, gRPC)
  • Familiarity with data engineering and building scalable data pipelines
  • Ability to work under pressure, manage deadlines, and exercise independent judgment

Citi Compensation & Benefits Highlights

The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about Citi and has not been reviewed or approved by Citi.

  • Healthcare Strength Benefits coverage is positioned as comprehensive, including health, dental, and vision insurance plus on-site clinics, prescription drug support, and disability coverage. Family-building support such as fertility assistance is described as a notable differentiator within the overall package.
  • Retirement Support Retirement benefits are framed as strong, highlighted by a 401(k) with matching and additional plan options like a Roth 401(k). Financial support is reinforced through discounts and broader financial guidance resources tied to the benefits ecosystem.
  • Wellbeing & Lifestyle Benefits Wellbeing support extends beyond insurance through programs like an Employee Assistance Program, counseling/legal resources, and gym or wellness reimbursement. These offerings increase the perceived total rewards value even when cash compensation sentiment varies by role.

Citi Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Kwun Tong, Kowloon
223,850 Employees

What We Do

Citi's mission is to serve as a trusted partner to our clients by responsibly providing financial services that enable growth and economic progress. Our core activities are safeguarding assets, lending money, making payments and accessing the capital markets on behalf of our clients. We have 200 years of experience helping our clients meet the world's toughest challenges and embrace its greatest opportunities. We are Citi, the global bank – an institution connecting millions of people across hundreds of countries and cities.

Similar Jobs

BetaCraft Technologies Logo BetaCraft Technologies

Machine Learning Engineer

Cloud • Information Technology • Professional Services • Software
In-Office
Pune, Mahārāshtra, IND
7 Employees

Canonical Logo Canonical

Software Engineer

Cloud • Software
In-Office or Remote
7 Locations
880 Employees
2K-2K Annually

TransUnion Logo TransUnion

.NET Developer

Big Data • Fintech • Information Technology • Business Intelligence • Financial Services • Cybersecurity • Big Data Analytics
Hybrid
2 Locations
13000 Employees

Optum Logo Optum

Software Engineering Lead - FSE

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Pune, Mahārāshtra, IND
160000 Employees

Similar Companies Hiring

Hanover Park Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
42 Employees
Kepler  Thumbnail
Fintech • Software
New York, New York
6 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account