Citi is looking for a mid-level AI/ML Engineer to focus on the implementation, optimization, and deployment of machine learning systems, including NLP and Generative AI solutions. You will build and deploy models, contribute to scalable platform design, and work collaboratively within a globally distributed team.
What You'll Do
- Work as a collaborative member of a team spread over multiple locations (India, UK, US).
- Adhere to internally published architectural guidelines and participate in architectural discussions and reviews.
- Contribute to defining and communicating development standards and participate in code reviews.
- Develop high-quality code and actively participate in design & code reviews.
- Assist in designing and developing scalable & configurable platform functionality for global applications.
- Contribute to the design and architecture of scalable machine learning systems, ensuring integration with existing platforms.
- Support the end-to-end lifecycle of deploying and operationalizing machine learning models in production environments.
- Develop, optimize, and implement machine learning algorithms and statistical models.
- Implement and integrate Generative AI (LLM) and Natural Language Processing (NLP) solutions, including prompting strategies, LLM-based data augmentation, and Retrieval-Augmented Generation (RAG).
- Participate in the implementation and adherence to MLOps practices, including CI/CD pipelines, automated testing, monitoring, and retraining strategies.
- Contribute to the optimization of ML models and underlying infrastructure for computational efficiency, speed, and resource utilization.
- Actively participate in technical discussions, adhere to best coding practices, and engage in code reviews.
- Partner with data scientists, product managers, and other engineering teams to translate business requirements into technical ML solutions.
- Adhere to risk assessment and compliance guidelines in ML system design and deployment.
What We're Looking For
- 3+ years of hands-on experience in Machine Learning Engineering, MLOps, or AI system development.
- Some experience or familiarity with Generative AI/LLM concepts and their application.
- Proficiency in Python for ML development, with experience in relevant libraries (scikit-learn, pandas, SpaCy, TensorFlow, PyTorch, Hugging Face Transformers).
- Experience with PySpark for large-scale data processing and building data pipelines.
- Familiarity with big data frameworks (Hadoop, Spark, Hive, Hue) and exposure to streaming technologies.
- Experience with MLOps tools and practices (e.g., Docker, Kubernetes, MLflow, Airflow, CI/CD for ML).
- Experience in implementing and deploying NLP and deep learning models.
- Familiarity with Generative AI development concepts, including prompting strategies and RAG implementation.
- Familiarity with cloud platforms (AWS, Azure, GCP) and their ML services.
- Understanding of scalable and performant ML system design principles.
- Strong analytical, interpretive, and problem-solving skills with a focus on engineering challenges.
- Good interpersonal, verbal, and written communication skills, with the ability to articulate technical concepts.
- Ability to work independently on assigned tasks, contribute to project completion, and a strong willingness to learn and grow.
Nice to Have
- Experience with frameworks like Langchain or LlamaIndex is a plus.
- Exposure to graph neural networks, graph databases, or distributed systems for ML.
Technical Stack
- Languages & Core ML: Python, scikit-learn, pandas, SpaCy, TensorFlow, PyTorch, Hugging Face Transformers
- Data & Big Data: PySpark, Hadoop, Spark, Hive, Hue
- MLOps & Infrastructure: Docker, Kubernetes, MLflow, Airflow
- Cloud Platforms: AWS, Azure, GCP
- AI Frameworks: Langchain, LlamaIndex
Team & Environment
You will be part of the Artificial Intelligence and Machine Learning team, spread over multiple locations (India, UK, US).
Benefits & Compensation
- Compensation: $96,960.00 - $145,440.00
- Medical, dental & vision coverage
- 401(k)
- Life, accident, and disability insurance
- Wellness programs
- Paid time off packages including vacation, sick leave, and paid holidays
- Discretionary and formulaic incentive and retention awards
Work Mode
This is a local-city role based in Tampa, Florida, United States.
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.
