Senior Lead Software Engineering - AI/ML Platform

JPMorgan Chase - Wilmington, DE

Full-time - Senior

Hybrid - Wilmington, DE

Credit Intermediation and Related Activities

About the position

As a Senior Lead Software Engineer at JPMorgan Chase within the Corporate Sector, AI/ML Technology, you will be a key member of an agile team focused on enhancing, building, and delivering market-leading ML technology products. This role involves architecting and implementing distributed ML infrastructure, optimizing system performance, and integrating Generative AI within the ML Platform. You will leverage your deep technical expertise to drive significant business impact and contribute to a culture of diversity, equity, and inclusion.

Responsibilities

• Architects and implements distributed ML infrastructure, including inference, training, scheduling, orchestration, and storage.
• Develops advanced monitoring and management tools for high reliability and scalability.
• Optimizes system performance by identifying and resolving inefficiencies and bottlenecks.
• Collaborates with product teams to deliver tailored, technology-driven solutions.
• Drives decisions that influence product design, application functionality, and technical operations and processes.
• Integrates Generative AI within the ML Platform using state-of-the-art techniques.
• Contributes to a team culture of diversity, equity, inclusion, and respect.

Requirements

• Formal training or certification on software engineering concepts and 5+ years applied experience.
• Deep expertise in AWS / Azure and Kubernetes ecosystem, including EKS, Helm, Custom Operators and Terraform.
• Advanced in Python programming language; Java is a plus.
• Background in High Performance Computing, ML Hardware Acceleration (e.g., GPU, TPU, RDMA), or ML for Systems.
• Strong coding skills and experience in developing large-scale ML systems.
• Extensive hands-on experience with ML frameworks (TensorFlow, PyTorch, JAX, scikit-learn).
• 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.
• Ability to tackle design and functionality problems independently with little to no oversight.

Nice-to-haves

• Excellent problem-solving and analytical skills.
• Ability to work independently and in a team.
• Passion for innovations and continuous learning.

Benefits

• Comprehensive health care coverage
• On-site health and wellness centers
• Retirement savings plan
• Backup childcare
• Tuition reimbursement
• Mental health support
• Financial coaching