Data Engineer - Python, AI

Citibank (Switzerland) AG; Pune District; Hybrid; Confidential; Full time

Job summary

Citibank (Switzerland) AG is seeking a mid-level Data Engineer with Python, NLP, and data engineering expertise. You will build NLP pipelines with Flair/BERT, process large data with PySpark and Pandas, and develop Flask APIs for model inference.

The role involves CI/CD, MLflow, Redis, Autosys, and AWS boto3 in a hybrid Pune setting. You will collaborate across teams, ensure scalable data processing, and contribute to deployment pipelines and monitoring.

Qualifications

• 10–12 years of hands-on Python programming experience.
• Strong fundamentals in Python, OOP, and design patterns.
• Experience with NLP libraries such as Flair, BERT, HuggingFace Transformers, or similar.
• Solid experience with PySpark, Pandas, PyArrow, and distributed data pipelines.
• Experience building APIs using Flask (FastAPI is a plus).
• Experience with MLflow for model tracking and deployment.
• Good understanding of CI/CD practices and Git workflows.
• Experience working with Redis or similar in-memory stores.
• Experience with Autosys JILs for job scheduling.
• Comfortable with Linux command line and shell scripting.
• Strong debugging, problem-solving, and teamwork skills.
• Exposure to cloud services; AWS boto3 experience is an asset.

Responsibilities

• Develop and optimize ETL/data processing jobs using PySpark, Pandas, PyArrow, and related libraries.
• Build and maintain NLP pipelines using Flair, BERT, and LLM-based models.
• Develop scalable ingestion and data transformation pipelines for AI and analytics use cases.
• Build and maintain Flask-based APIs for model inference and service integrations.
• Use regular expressions for text cleaning, parsing, and NLP preprocessing.
• Integrate caching and fast lookups using Redis.
• Manage and deploy ML models using MLflow for tracking and versioning.
• Support CI/CD workflows using GitHub, LightSpeed Enterprise, and deployment pipelines.
• Create and maintain Autosys JILs for job scheduling and automation.
• Use basic Linux commands for troubleshooting, operations, and deployment tasks.
• Monitor application and system health using ITRS Geneos.
• Write unit tests and improve automation test coverage (PyTest/unittest).
• Work with REST APIs, microservices, and basic shell scripting.

Skills

Python programming

NLP

Distributed data pipelines

Debugging

Teamwork

CI/CD practices

Linux CLI

REST APIs

Model deployment

Tools

PySpark

Pandas

PyArrow

Flask

MLflow

GitHub

LightSpeed Enterprise

Autosys

boto3

Redis

Docker

Kubernetes

Airflow

Job description

## Data Engineer - Python, AIApplyremote type: Hybridlocations: Pune Maharashtra Indiatime type: Full timeposted on: Posted Todaytime left to apply: End Date: June 18, 2026 (13 days left to apply)job requisition id: 26965263Role Summary We are looking for a mid-level Python Developer with combined experience in Data Engineering and AI/NLP engineering. The candidate will build NLP pipelines using libraries such as Flair, BERT, and LLM frameworks, and will also work on large-scale data processing using PySpark, Pandas, and related data tools. The role includes developing APIs, integrating with platform services, and supporting CI/CD deployments using GitHub and LightSpeed Enterprise.## Key Responsibilities* Develop and optimize ETL/data processing jobs using PySpark, Pandas, PyArrow, and related libraries.* Build and maintain NLP pipelines using Flair, BERT, and LLM-based models.* Develop scalable ingestion and data transformation pipelines for AI and analytics use cases.* Build and maintain Flask-based APIs for model inference and service integrations.* Use regular expressions for text cleaning, parsing, and NLP preprocessing.* Integrate caching and fast lookups using Redis.* Manage and deploy ML models using MLflow for tracking and versioning.* Support CI/CD workflows using GitHub, LightSpeed Enterprise, and deployment pipelines.* Create and maintain Autosys JILs for job scheduling and automation.* Use basic Linux commands for troubleshooting, operations, and deployment tasks.* Monitor application and system health using ITRS Geneos.* Write unit tests and improve automation test coverage (PyTest/unittest).* Work with REST APIs, microservices, and basic shell scripting.* Work with cloud services (ECS), including boto3.## Required Skills* 10–12 years of hands-on Python programming experience.* Strong fundamentals in Python, OOP, and design patterns.* Experience with NLP libraries such as Flair, BERT, HuggingFace Transformers, or similar.* Solid experience with PySpark, Pandas, PyArrow, and distributed data pipelines.* Experience building APIs using Flask (FastAPI is a plus).* Experience with MLflow for model tracking and deployment.* Good understanding of CI/CD practices and Git workflows.* Experience working with Redis or similar in-memory stores.* Experience with Autosys JILs for job scheduling.* Comfortable with Linux command line and shell scripting.* Strong debugging, problem-solving, and teamwork skills.* Exposure to cloud services; AWS boto3 experience is an asset.## Nice-to-Have* Experience with Polars or Dask for high-performance data processing.* Experience with PyTorch or TensorFlow for model training.* Experience with Docker, Kubernetes, or containerized deployments.* Experience with monitoring tools such as ITRS Geneos.* Experience with FastAPI, Airflow, or Prefect.------------------------------------------------------## **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.*