Lead Software Engineer

Join our team and lead the charge in transforming production stability at JPMorgan Chase, where your problem-solving expertise drives innovation and reliability in a collaborative, high-impact environment.

As a Lead Software Engineer at JPMorgan Chase within the AI Platform, you will work with cutting-edge technologies and keep up with the latest AI coding techniques. You will lead teams of developers, teaching them best practices in high-performance computing (HPC). You will collaborate with cross-functional teams to build and deploy machine learning at a global scale. You will identify and harness various accelerators to enable machine learning and AI across all JPMorgan’s businesses. You will deliver software that outperforms other platforms, serving our various business needs. You will measure the capabilities that you create, providing a quantified explanation of your benefits and tradeoff choices. You will build a world-class machine learning platform using data-promoten decisions that efficiently leverage today’s and future HPC architectures.

You will play a crucial role in building and deploying machine learning at the scale of a global financial leader.  You will help identify, evaluate and harness various accelerators to enable business applications involving machine learning and AI across all lines of JPMorgan’s businesses.  You will infuse the JPMorgan developer community with an appreciation of the impact that HPC can have by delivering software that consistently outperforms other platforms.  You will deliver a variety of options to serve our various business needs--sometimes driven by low-latency; other times driven by throughput or low power.  You find it instinctual to measure the capabilities that you create, providing a quantified explanation of your benefits and tradeoff choices.

In short, you are thrilled by the prospect of building a world-class machine learning platform using data-driven decisions that takes advantage of today’s accelerator-driven HPC architectures efficiently, while also positioning JPMorgan to leverage many future architectures.

Job responsibilities

• Build scalable and efficient inferencing and training pipelines using HPC software techniques and patterns
• Working closely with business and data science teams, develop easy-to-use systems that serve their needs
• Using telemetry, create measurable frameworks for deciding amongst hardware and software options
• Publish and support re-usable patterns to optimize training and inference of ML models on various architectures
• Support developer community in learning lessons from high-performance computing (HPC) domain

Required qualifications, capabilities and skills

• "Formal training or certification on High-Performance Computing system architectures and network topologies concepts and 5+ years applied experience
• Advanced understanding of High-Performance Computing system architectures and network topologies
• Expertise in at least one accelerator type (e.g., GPU, FPGA) and experience mapping LLMs onto these accelerators
• Proficiency parallel programming and performance analysis of accelerator-based systems
• Strong programming skills in Python, scripting, C, C++ with experience in AI/ML frameworks like PyTorch and LangChain
• Familiarity with HPC software (e.g., NCCL, MPI) and resource schedulers (e.g., Kubernetes, SLURM).
• Excellent communication skills and the ability to work collaboratively in a dynamic team environment.

Experience:

• Master’s Degree in Computer Science (required)
• 5+ years of experience in high-performance computing software
• 3+ years of experience with accelerators and deep learning, particularly large language model

Why Join Us?

• Be at the forefront of AI innovation in the financial sector
• Collaborate with a diverse team of experts and contribute to impactful projects
• Enjoy opportunities for professional growth and leadership development

If you're passionate about building world-class machine learning platforms and excited by the challenges of accelerator-driven HPC architectures, we invite you to apply and be part of our journey to redefine AI at JPMorgan Chase.