Applied AI ML Director, Principal Machine Learning Platform Engineer

Posted 2 Days Ago
Be an Early Applicant
San Francisco, CA
Hybrid
Expert/Leader
Financial Services
The Role
The Applied AI ML Director will lead the development of ML infrastructure and tools, enhance model performance, mentor engineers, and align ML projects with business goals, ensuring seamless collaboration across teams.
Summary Generated by Built In

Job Description
Job Description
We are seeking a highly skilled and innovative AI ML Director, Principal Machine Learning Platforms to join our team within the Corporate AI ML Technology Group. The ideal candidate will have extensive experience in traditional AI, ML infrastructure , ML Platform tools , GenAI, and Machine Learning Platforms.
As an Executive Director, Applied AI/ML, Machine Learning Platforms, 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.
In this role, you will:

  • Achieve state-of-the-art throughput for critical models using advanced techniques like model parallelism and distributed training.
  • Reduce inference time for new model architectures using optimizations like quantization and pruning.
  • Collaborate closely with Applied AI engineering to optimize the internal inference stack, leveraging technologies like TensorRT, ONNX, etc.
  • Recruit and mentor top-tier AI systems engineers, fostering a culture of continuous learning and innovation.
  • Coordinate the inference needs of JPMC's research teams, ensuring alignment with business goals.


Job responsibilities:

  • Architect and implement distributed ML infrastructure, including inference, training, scheduling, orchestration, and storage.
  • Develop advanced monitoring and management tools for high reliability and scalability.
  • Optimize system performance by identifying and resolving inefficiencies and bottlenecks.
  • Collaborate with product teams to deliver tailored, technology-driven solutions.
  • Drive the adoption and execution of ML Platform tools across various teams.
  • Integrate Generative AI within the ML Platform using state-of-the-art techniques


Required Qualifications, Capabilities, and Skills:

  • 10+ years of experience in engineering management with a strong technical background.
  • Extensive hands-on experience with ML frameworks (TensorFlow, PyTorch, JAX, scikit-learn).
  • Deep expertise in AWS / GCP and Kubernetes ecosystem, including EKS, Helm, and custom operators.
  • Strong coding skills and experience in developing large-scale ML systems.
  • Background in High Performance Computing, ML Hardware Acceleration (e.g., GPU, TPU, RDMA), or ML for Systems.
  • Proven track record in contributing to and optimizing open-source ML frameworks.
  • Strategic thinker with the ability to craft and drive a technical vision for maximum business impact.
  • Demonstrated leadership in working effectively with engineers, data scientists, and ML practitioners.
  • Proven ability to identify trade-offs, clarify project ambiguities, and drive decision-making.


About Us
JPMorganChase, 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 & Co. 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.

Top Skills

Jax
PyTorch
Scikit-Learn
TensorFlow

What the Team is Saying

Nick S.
Lupe C.
Edwin T.
Dawn T.
Meng M.
The Company
San Francisco, CA
250,000 Employees
Hybrid Workplace

What We Do

JPMorgan Chase & Co. (NYSE: JPM) is a leading global financial services firm with assets of $3.7 trillion and operations worldwide. The firm is a leader in investment banking, financial services for consumers and small businesses, commercial banking, financial transaction processing, and asset management. A component of the Dow Jones Industrial Average, JPMorgan Chase & Co. serves millions of consumers in the United States and many of the world’s most prominent corporate, institutional and government clients under its J.P. Morgan and Chase brands.

Technology fuels every aspect of our company and is at the heart of everything we do. With over 50,000 technologists globally and an annual tech spend of $12 billion, we are dedicated to improving the design, analytics, development, coding, testing and application programming that goes into creating high quality software and new products.

Learn more about technology at our firm, explore resources from our Distinguished Engineers, AI & ML researchers, and other experts; access the latest episode of our TechTrends podcast, and more at www.jpmorgan.com/technology. Information about JPMorgan Chase & Co. is available at www.jpmorganchase.com.

©2023 JPMorgan Chase & Co. All rights reserved. JPMorgan Chase is an Equal Opportunity Employer, including Disability/Veterans.

Why Work With Us

Our technologists work on a diverse range of solutions that include strategic technology initiatives, big data, mobile, electronic payments, machine learning, cybersecurity, enterprise cloud development, and other state-of-the-art technologies.

JPMorganChase Teams

Team
CONTINUOUS LEARNING AND INNOVATION
About our Teams

JPMorganChase Offices

Hybrid Workspace

Employees engage in a combination of remote and on-site work.

Typical time on-site: Flexible
San Francisco, CA

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account