About the Company
Working at Citi is far more than just a job. A career with us means joining a team of more than 230,000 dedicated people from around the globe. At Citi, you'll have the opportunity to grow your career, give back to your community and make a real impact.
Responsibilities
- Design, Develop and deploy generative models (e.g., GPT, GAN, VAEs) for various applications
- Customize and fine-tune existing RAG frameworks or design new RAG to meet project requirements
- Collaborate with cross-functional team to understand business requirements and translate them into AI solutions
- Conduct research to advance the state-of-the art in generative modeling and stay up to date with latest advancements in the field
- Optimize and fine-tune models for performance, scalability, and robustness
- Implement and maintain AI pipelines and infrastructure to support model training and deployment
- Perform data analysis and preprocessing to ensure high-quality input for model training
- Mentor junior team members and provide technical guidance
- Write and maintain comprehensive documentation for models and algorithms
- Present findings and project progress to stakeholders and management
Requirements
- Bachelors or Masters degree in computer science, Data Science, Machine Learning, or a related field. A Ph.D. is a plus
- 8+ years of experience in machine learning and deep learning, with a focus on generative models
- Strong proficiency in Python and deep learning frameworks such as TensorFlow, PyTorch, or Keras
- Experience with natural language processing (NLP) and natural language generation (NLG)
- Proven track record of building and deploying generative models in production environments particularly using RAG frameworks
- Solid understanding of machine learning algorithms, data structures, and software engineering principles
- Excellent problem-solving skills and the ability to work independently and as part of a team
- Strong communication skills, with the ability to explain complex technical concepts to non-technical stakeholders