Senior Lead Software Engineer - AI/ML Platform Architect

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

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