Senior Lead Software Engineer - AI/ML Developer

Socket.dev; New Jersey; On-site; USD 180,000 - 240,000; Full time

Job summary

JPMorgan Chase in the United States is seeking a Lead Software Engineer - AI/ML Solutions within the AI/Machine Learning Platform Engineering Team. You will design and deliver trusted infrastructure and tools that empower data scientists to develop, deploy, and monitor models securely and efficiently.

You will design scalable ML platforms, optimize tooling for training, deployment, monitoring, and lifecycle management, and integrate data engineering, feature stores, and model serving into

Qualifications

• Formal training or certification on software engineering concepts.
• Hands-on experience building, deploying, and maintaining ML platforms or infrastructure.
• Advanced Python and ML frameworks (TensorFlow, PyTorch, Scikit-learn).

Responsibilities

• Designs, builds, and maintains scalable ML platforms and infrastructure to support end-to-end ML workflows.
• Develops and optimizes tools for model training, deployment, monitoring, and lifecycle management.
• Integrates data engineering, feature management, and model serving capabilities into unified ML platform solutions.
• Implements secure, high-quality production code for platform services, APIs, and automation pipelines.
• Leads evaluation sessions with data scientists, ML engineers, and product teams to understand requirements and deliver platform features that accelerate ML development and operations.
• Ensures platform reliability, scalability, and performance through proactive monitoring, troubleshooting, and continuous improvement.
• Produces architecture and design artifacts for platform components, ensuring alignment with enterprise standards and best practices.
• Automates infrastructure provisioning, configuration, and CI/CD pipelines for ML platform services.
• Contributes to the ML platform engineering community of practice and participate in events that explore new and emerging technologies
• Drives team adoption of enterprise-authorized AI-assisted engineering practices within the work environment to improve code quality, delivery speed, and operational outcomes (e.g., AI-assisted code review/refactoring, test strategy acceleration, incident/root-cause analysis support), while establishing consistent validation standards (secure coding, peer review, automated testing) and promoting reuse of effective patterns across the team.
• Applies knowledge of tools within the Software Development Life Cycle toolchain, including enterprise-authorized AI-assisted development and automation capabilities, to improve the value realized by automation.

Skills

Python

ML frameworks

CI/CD for ML

APIs & microservices

Agile methods

Cross-functional collaboration

Education

Software engineering training/certification

Tools

Databricks

Snowflake

Docker

Kubernetes

Terraform

Job description

We have an opportunity to impact your career and provide an adventure where you can push the limits of what's possible.

As a Lead Software Engineer - AI/ML Solutions at JPMorgan Chase within the Consumer & Community Banking's AI/Machine Learning Platform Engineering Team, you serve as a seasoned member of an agile team focused on building, scaling, and maintaining robust machine learning platforms. You will design and deliver trusted, market-leading infrastructure and tools that empower data scientists and ML engineers to develop, deploy, and monitor models efficiently and securely. You are responsible for implementing critical technology solutions across multiple technical areas to support the firm’s business objectives and drive innovation in ML platform capabilities.

Job responsibilities

• Designs, builds, and maintains scalable machine learning platforms and infrastructure to support end-to-end ML workflows.
• Develops and optimizes tools for model training, deployment, monitoring, and lifecycle management.
• Integrates data engineering, feature management, and model serving capabilities into unified ML platform solutions.
• Implements secure, high-quality production code for platform services, APIs, and automation pipelines.
• Leads evaluation sessions with data scientists, ML engineers, and product teams to understand requirements and deliver platform features that accelerate ML development and operations.
• Ensures platform reliability, scalability, and performance through proactive monitoring, troubleshooting, and continuous improvement.
• Produces architecture and design artifacts for platform components, ensuring alignment with enterprise standards and best practices.
• Automates infrastructure provisioning, configuration, and CI/CD pipelines for ML platform services.
• Contributes to the ML platform engineering community of practice and participate in events that explore new and emerging technologies
• Drives team adoption of enterprise-authorized AI-assisted engineering practices within the work environment to improve code quality, delivery speed, and operational outcomes (e.g., AI-assisted code review/refactoring, test strategy acceleration, incident/root-cause analysis support), while establishing consistent validation standards (secure coding, peer review, automated testing) and promoting reuse of effective patterns across the team.
• Applies knowledge of tools within the Software Development Life Cycle toolchain, including enterprise-authorized AI-assisted development and automation capabilities, to improve the value realized by automation.

Required qualifications, capabilities, and skills

• Formal training or certification on software engineering concepts and 5+ years applied experience
• Hands-on experience building, deploying, and maintaining machine learning platforms or infrastructure
• Advanced in Python and one or more ML frameworks (e.g., TensorFlow, PyTorch, Scikit-learn)
• Experience with data processing frameworks and tools (e.g., Spark, Pandas, SQL)
• Practical experience with cloud-based ML platforms (e.g., AWS SageMaker, GCP AI Platform, Azure ML) or on-prem ML infrastructure
• Strong understanding of MLOps practices, including CI/CD for ML, model versioning, and monitoring
• Experience developing APIs and platform services for ML workflows
• Proficient in all aspects if the Software Development Life Cycle and Agile Methodologies
• Ability to collaborate with cross-functional teams to deliver platform solutions aligned with business objectives
• Drives team adoption of enterprise-authorized AI-assisted engineering practices within the work environment to improve code quality, delivery speed, and operational outcomes (e.g., AI-assisted code review/refactoring, test strategy acceleration, incident/root-cause analysis support), while establishing consistent validation standards (secure coding, peer review, automated testing) and promoting reuse of effective patterns across the team.
• Applies knowledge of tools within the Software Development Life Cycle toolchain, including enterprise-authorized AI-assisted development and automation capabilities, to improve the value realized by automation.

Preferred qualifications, capabilities, and skills

• Familiarity with Databricks for scalable data engineering and ML platform integration
• Experience working with Snowflake for cloud-based data warehousing and analytics
• Exposure to Snorkel AI for programmatic data labeling and training data management
• Experience with containerization and orchestration tools (e.g., Docker, Kubernetes, Airflow)
• Familiarity with feature stores, model registries, and ML metadata management
• Experience with infrastructure-as-code tools (e.g., Terraform, CloudFormation)
• Experience with RESTful APIs and microservices architectures

FEDERAL DEPOSIT INSURANCE ACT:

This position is subject to Section 19 of the Federal Deposit Insurance Act. As such, an employment offer for this position is contingent on JPMorganChase’s review of criminal conviction history, including pretrial diversions or program entries.

Chase is a leading financial services firm, helping nearly half of America’s households and small businesses achieve their financial goals through a broad range of financial products. Our mission is to create engaged, lifelong relationships and put our customers at the heart of everything we do. We also help small businesses, nonprofits and cities grow, delivering solutions to solve all their financial needs.

We offer a competitive total rewards package including base salary determined based on the role, experience, skill set and location. Those in eligible roles may receive commission-based pay and/or discretionary incentive compensation, paid in the form of cash and/or forfeitable equity, awarded in recognition of 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 ourFAQsfor more information about requesting an accommodation.

Equal Opportunity Employer/Disability/Veterans

Our Consumer & Community Banking division serves our Chase customers through a range of financial services, including personal banking, credit cards, mortgages, auto financing, investment advice, small business loans and payment processing. We’re proud to lead the U.S. in credit card sales and deposit growth and have the most-used digital solutions – all while ranking first in customer satisfaction.