Applied AI ML Director

Job Description
We are seeking individuals with advanced expertise in Machine Learning (ML) to join our dynamic team.
As an Applied AI ML Director within our Corporate Sector, you will play a pivotal role in collaborating with various teams to develop and implement common ML components for AIT. You will contribute to our innovative projects and drive the future of machine learning at AIT.
Job 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 AutoML methods.
• Enhance ML workflows through advanced proficiency in transfer learning and pretrained models.

Required qualifications, capabilities, and skills

• MS and/or PhD in Computer Science, Machine Learning, or a related field, with at least five years of applied machine learning experience.
• Solid background in Natural Language Processing (NLP) and/or hands-on experience with machine learning and deep learning methods.
• Deep understanding and expertise in PyTorch/TensorFlow internals.
• Specialized experience in GPU optimization, including defining and implementing algorithms such as DeepSpeed, Megatron, Flash-Attention, and Grouped Attention.

Preferred qualifications, capabilities, and skills

• Experience with JAX, Ray, MLFlow, and/or other distributed training frameworks.
• In-depth understanding of Search/Ranking, Recommender systems, RAG (Similarity Search), Graph techniques, and other advanced methodologies.
• Advanced knowledge in Reinforcement Learning or Meta Learning.
• Understanding of Large Language Model (LLM) techniques, including Agents, Planning, Reasoning, and other related methods.
• Expertise in training and fine-tuning large language models (LLMs) and embedding models.

About Us
JPMorgan Chase & Co., one of the oldest financial institutions, offers innovative financial solutions to millions of consumers, small businesses and many of the world's most prominent corporate, institutional and government clients under the J.P. Morgan and Chase brands. Our history spans over 200 years and today we are a leader in investment banking, consumer and small business banking, commercial banking, financial transaction processing and asset management.
We offer a competitive total rewards package including base salary determined based on the role, experience, skill set, and location. For those in eligible roles, we offer discretionary incentive compensation which may be awarded in recognition of firm performance and individual achievements and contributions. We also offer a range of benefits and programs to meet employee needs, based on eligibility. These benefits include comprehensive health care coverage, on-site health and wellness centers, a retirement savings plan, backup childcare, tuition reimbursement, mental health support, financial coaching and more. Additional details about total compensation and benefits will be provided during the hiring process.
We recognize that our people are our strength and the diverse talents they bring to our global workforce are directly linked to our success. We are an equal opportunity employer and place a high value on diversity and inclusion at our company. We do not discriminate on the basis of any protected attribute, including race, religion, color, national origin, gender, sexual orientation, gender identity, gender expression, age, marital or veteran status, pregnancy or disability, or any other basis protected under applicable law. We also make reasonable accommodations for applicants' and employees' religious practices and beliefs, as well as mental health or physical disability needs. Visit our FAQs for more information about requesting an accommodation.
JPMorgan Chase is an Equal Opportunity Employer, including Disability/Veterans
About the Team
Our professionals in our Corporate Functions cover a diverse range of areas from finance and risk to human resources and marketing. Our corporate teams are an essential part of our company, ensuring that we're setting our businesses, clients, customers and employees up for success.