Data Engineer - Python, AI

Citi · citibenefits.com

Pune Maharashtra India Full-time yesterday
Workplace
On-site
Location
Pune Maharashtra India
Who can apply
Based in India: you usually need the right to work there

Check your CV against this job

Free · No signup · A 0–100 ATS match score and the keywords you're missing.

Check my CV free

$14.99/month, cancel anytime. Already have an account? Log in

About the role

Role 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.

Get matched & apply with FindAJobAI

Upload your resume once. We score every job against your profile, tailor your resume and cover letter, and autofill the application.