AI/ML Engineer

Responsibilities

• Build REST endpoints Refine prompts and write Agentic code Adjusting LLM Inference parameters Writing and refining prompts Selecting context for given LLM window size and setup Debugging Agent Self-Reflection loops Calling external search libraries and writing formatting code to connect prompt templates Writing code in compliance with risk restrictions Documenting code to ensure compliance and reusability of work Write SQL queries and optimize queries for LLM context retrieval Writing unit and integration tests for LLMs Qualifications: AI ML Engineer 6+ years of total experience 3+ Years experience as a data scientist
• machine learning engineer (data-adjacent profession) Proficient in every aspect of the data pipeline, including: Data cleaning and preprocessing Model Selection Feature engineering Model training and retraining Model evaluation Moving models into productions Building APIs Python expertise Familiar with FastAPI, Flask Some python library or libraries for calling LLMs (LangChain or similar) Data manipulation libraries (Pandas, Numpy, Scipy) NLPText processing and parsing library experience (SpaCy, NLTK, TextBlob Deep LearningFamiliarity with model binaries and inferencing, experience with HuggingFace, PyTorch
• Tensorflow a plus LLMs Able to write promptsUnderstanding of the fundamentals of Language Modeling SQL Education: Bachelor’s degree/University degree or equivalent experience ------------------------------------------------------ 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
• 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
• 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.