Tampa, United States of America Hybrid USD 96,960 - 145,440 Yearly

Citi is hiring an AI/ML Engineer

About the Role

Role OverviewWe are seeking a mid-level AI/ML Engineer to join a growing team focused on advancing artificial intelligence and machine learning capabilities. This role involves designing, implementing, and maintaining intelligent systems that scale across global platforms, with a strong focus on NLP and Generative AI technologies.

Key Responsibilities

  • Collaborate with engineering teams across India, the UK, and the US in a distributed environment
  • Follow and contribute to architectural standards, development practices, and code quality initiatives
  • Design and build scalable, configurable components for enterprise-wide applications
  • Develop and refine machine learning models with attention to performance, reliability, and maintainability
  • Implement and integrate NLP and large language model (LLM) solutions using advanced techniques such as Retrieval-Augmented Generation (RAG)
  • Design prompting strategies and data augmentation methods tailored for LLMs
  • Support full lifecycle deployment of ML models, including monitoring, retraining, and version control
  • Apply MLOps principles using CI/CD pipelines, automated testing, and model tracking tools
  • Optimize models and infrastructure for efficiency, speed, and resource utilization
  • Work closely with data scientists and product teams to translate business needs into technical implementations
  • Ensure compliance with risk, regulatory, and internal policy requirements in all stages of development

Required Qualifications

  • Minimum of 3 years of hands-on experience in machine learning engineering, MLOps, or AI systems development
  • Strong proficiency in Python and ML libraries such as scikit-learn, pandas, SpaCy, TensorFlow, PyTorch, and Hugging Face Transformers
  • Experience with PySpark and building large-scale data pipelines
  • Familiarity with big data platforms including Hadoop, Spark, Hive, and Hue, as well as streaming technologies
  • Proven experience with MLOps tools such as Docker, Kubernetes, MLflow, and Airflow
  • Background in deploying NLP and deep learning models in production environments
  • Understanding of Generative AI concepts, including prompting and RAG architectures
  • Solid grasp of scalable system design for machine learning applications
  • Strong problem-solving abilities and analytical thinking
  • Effective communication skills, with the ability to explain technical concepts clearly
  • Self-motivated with a commitment to continuous learning and improvement

Preferred Qualifications

  • Experience with Langchain or LlamaIndex frameworks
  • Knowledge of cloud platforms (AWS, Azure, GCP) and their machine learning services
  • Exposure to graph neural networks, graph databases, or distributed computing systems

Work Environment

This is a hybrid role based in Tampa, Florida, with regular collaboration across global teams in India, the UK, and the US. The position supports flexible, cross-regional coordination within a structured technical framework.

Technical Stack

Python, scikit-learn, pandas, SpaCy, TensorFlow, PyTorch, Hugging Face Transformers, PySpark, Hadoop, Spark, Hive, Hue, Docker, Kubernetes, MLflow, Airflow, CI/CD, Langchain, LlamaIndex, AWS, Azure, GCP

Compensation & Benefits

Base salary range: $96,960 – $145,440 annually. Additional discretionary and formula-based incentive awards may apply. Benefits include comprehensive medical, dental, and vision coverage; 401(k) with company match; life, accident, and disability insurance; wellness programs; and paid time off including vacation, sick leave, and holidays.

Company Culture

We foster a collaborative, inclusive workplace that values technical excellence and continuous growth. Our team operates with integrity, embraces diverse perspectives, and adheres to high standards of compliance and ethical conduct. We are committed to equal opportunity and welcome applicants from all backgrounds.

Required Skills
Pythonscikit-learnpandasSpaCyTensorFlowPyTorchHugging Face TransformersPySparkHadoopSparkMachine Learning EngineeringMLOpsGenerative AILLMHive Pythonscikit-learnpandasSpaCyTensorFlowPyTorchHugging Face TransformersPySparkHadoopSparkMLOpsDockerKubernetesMLflowAirflow
Relocating to Thailand?

Visa and work permit handled by experts

SVBL manages your entire visa process — from application to approval. Work permits, extensions, and compliance all covered. One partner for legal, immigration, and settling in.

Work permit processing
Visa extensions & renewals
Immigration compliance
Banking & housing guidance
Get free consultation
Free initial consultation
About company
Citi
A global financial institution serving clients worldwide, providing financial services and solutions with a focus on responsible banking and economic progress.
All jobs at Citi Visit website
Job Details
Department Software Development
Category data
Posted 2 months ago