Applied AI ML Associate

About the Role

Bring your applied research and machine learning skills to a team building modern, high-impact generative AI capabilities. You'll work in a collaborative environment where experimentation, engineering excellence, and customer focus come together to deliver real outcomes. As an Applied AI ML Associate at JPMorganChase within the Applied Artificial Intelligence and Machine Learning Data Platforms team in Corporate Sector, you will help design, develop, and productionize machine learning and deep learning solutions, including generative AI use cases. You will partner closely with product and engineering stakeholders to turn business needs into reliable models, workflows, and platform capabilities. You will contribute to a scalable platform approach that can support multiple business lines while maintaining strong standards for performance, quality, and usability.

Key Responsibilities

• Serve as a subject matter expert on a wide range of ML techniques and optimizations.
• Provide in-depth knowledge of ML algorithms, frameworks, and techniques.
• Enhance ML workflows through advanced proficiency in Large Language Models (LLMs) and related techniques.
• Conduct experiments using latest ML technologies, analyzing results, tuning models.
• Provide hands-on coding to bring experimental results into production solutions by collaborating with the engineering team.
• Own end-to-end code development in Python for both proof of concept/experimentation and production-ready solutions.
• Optimize system accuracy and performance by identifying and resolving inefficiencies and bottlenecks.
• Collaborate with product and engineering teams to deliver tailored, science and technology-driven solutions.
• Integrate Generative AI within the ML Platform using state-of-the-art techniques.

Requirements & Tech Stack

• MS and/or PhD in Computer Science, Machine Learning, or a related field.
• Experience in one of the programming languages like Python, Java, C/C++.
• Intermediate Python is a must.
• Experience in applying data science, ML techniques to solve business problems.
• Solid background in Natural Language Processing (NLP) and Large Language Models (LLMs).
• Hands-on experience with machine learning and deep learning methods.
• Deep understanding and expertise in deep learning frameworks such as PyTorch or TensorFlow.
• Experience in advanced applied ML areas such as GPU optimization, fine tuning, embedding models, inferencing, prompt engineering, evaluation, RAG (Similarity Search).
• Ability to work on tasks and projects through to completion with limited supervision.
• Passion for detail and follow-through.
• Excellent communication skills and team player.

Preferred Qualifications

• Master's degree in computer science, ML or related areas.
• Experience with Ray, MLFlow, and/or other distributed training frameworks.
• In-depth understanding of Search/Ranking, Recommender systems, Graph techniques, and other advanced methodologies.
• Deep understanding of Large Language Model (LLM) techniques, including Agents, Planning, Reasoning, and other related methods.
• Experience with building and deploying ML models on cloud platforms such as AWS and AWS tools like Sagemaker, EKS, etc.
• Experience working with large-scale MLOps pipelines, working with and deploying models to production services.