Lead Software Engineer – AI/ML Data Platform (Test Engineering)
JPMorgan Chase & Co.About the role
Job Description
At J.P. Morgan Chase, we are building an enterprise-grade AI/ML Data Platform that enables scalable, secure, and responsible machine learning across the firm. We are seeking a Lead Software Engineer with a strong background in software development and platform engineering to help drive the Test Engineering Program—a strategic initiative focused on building robust, intelligent validation frameworks and infrastructure that power our ML and data products.
This is a software engineering role embedded in quality and reliability initiatives. You’ll architect and develop tools, services, and automation that elevate platform assurance across large-scale distributed systems. If you're passionate about platform quality, CI/CD excellence, infrastructure-as-code, and bringing engineering rigor to validation, this is your opportunity to make a major impact.
Key Responsibilities
• Design and build high-performance tools and services to validate the reliability, performance, and correctness of ML data pipelines and AI infrastructure.
• Develop platform-level test solutions and automation frameworks using Python, Terraform, and modern cloud-native practices.
• Contribute to the platform’s CI/CD pipeline by integrating automated testing, resilience checks, and observability hooks at every stage.
• Lead initiatives that drive testability, platform resilience, and validation as code across all layers of the ML platform stack.
• Collaborate with engineering, MLOps, and infrastructure teams to embed quality engineering deeply into platform components.
• Build reusable components that support scalability, modularity, and self-service quality tooling.
• Mentor junior engineers and influence technical standards across the Test Engineering Program.
Required Qualifications
• Bachelor’s or Master’s degree in Computer Science, Engineering, or a related technical field.
• 5+ years of hands-on software development experience, including large-scale backend systems or platform engineering.
• Expert in Python with a strong understanding of object-oriented programming, testing frameworks, and automation libraries.
• Experience building or validating platform infrastructure, with hands-on knowledge of CI/CD systems, GitHub Actions, Jenkins, or similar tools.
• Solid experience with AWS services (Lambda, S3, ECS/EKS, Step Functions, CloudWatch).
• Proficient in Infrastructure as Code using Terraform to manage and provision cloud infrastructure.
• Strong understanding of software engineering best practices: code quality, reliability, performance optimization, and observability.
Preferred Qualifications
• Exposure to machine learning workflows, model lifecycle management, or data engineering platforms.
• Experience with distributed systems, event-driven architectures (e.g., Kafka), and big data platforms (e.g., Spark, Databricks).
• Familiarity with banking or financial domain use cases, including data governance and compliance-focused development.
• Knowledge of platform security, monitoring, and resilient architecture patterns.
Why Join Us
• Help shape the test engineering strategy for one of the largest AI/ML data platforms in the financial industry.
• Collaborate with highly skilled engineers and data scientists in a high-impact, growth-oriented environment.
• Gain access to world-class infrastructure, tooling, and learning opportunities.
• Build software that powers decisions for millions of customers and drives real business value.
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 ca
Apply for this role
Generate a tailored application kit with a matched cover letter, interview prep, and CV highlights — in under 60 seconds.
Apply Now →Generate Application KitFree account required — sign up in 30s