Software Engineer II - Machine Learning Platform

JPMorgan Chase & Co. • Palo Alto, CA

19h

About The Position

Advance your career by contributing to the development of robust machine learning platforms at JPMorgan Chase, gaining valuable skills and experience as you help drive innovative solutions for data scientists and ML engineers.
As a Software Engineer II at JPMorgan Chase in the Consumer and Community Banking Technology team, you will be an integral part of an agile engineering team focused on enhancing, designing, and delivering secure, stable, and scalable technology products. Leveraging your technical skills and eagerness to learn, you will support the development and maintenance of infrastructure and tools that enable efficient model development, deployment, and monitoring. Working alongside senior engineers, you will deepen your expertise in ML platform engineering, learn best practices, and help shape the future of machine learning at JPMorgan Chase.

Requirements

• 2+ years of applied experience or formal training/certification in software engineering concepts.
• Hands-on practical experience in system design, application development, testing, and operational stability.
• Experience in developing, debugging, and maintaining code in a large corporate environment with one or more modern programming languages and database querying languages.
• Proficiency in Python and familiarity with at least one ML framework (e.g., TensorFlow, PyTorch, Scikit-learn).
• Experience with data processing tools (e.g., Spark, Pandas, SQL).
• Exposure to cloud-based ML platforms (e.g., AWS SageMaker, GCP AI Platform, Azure ML) or on-prem ML infrastructure.
• Understanding of MLOps practices, including CI/CD for ML, model versioning, and monitoring.
• Experience developing APIs or platform services for ML workflows.
• Knowledge of the software development life cycle and agile methodologies.
• Ability to collaborate with cross-functional teams to deliver platform solutions aligned with business objectives.

Nice To Haves

• Familiarity with Databricks for scalable data engineering and ML platform integration.
• Experience 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.

Responsibilities

• Assist in designing, building, and maintaining scalable machine learning platforms and infrastructure to support end-to-end ML workflows.
• Develop and enhance tools for model training, deployment, monitoring, and lifecycle management under the guidance of senior engineers.
• Support the integration of data engineering, feature management, and model serving capabilities into unified ML platform solutions.
• Write secure, high-quality production code for platform services, APIs, and automation pipelines.
• Collaborate with data scientists, ML engineers, and product teams to understand requirements and deliver platform features that accelerate ML development and operations.
• Participate in ensuring platform reliability, scalability, and performance through monitoring, troubleshooting, and continuous improvement.
• Contribute to architecture and design documentation for platform components, ensuring alignment with enterprise standards.
• Help automate infrastructure provisioning, configuration, and CI/CD pipelines for ML platform services.
• Engage with the ML platform engineering community and participate in events to learn about new and emerging technologies.

Benefits

• 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.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Education Level: No Education Listed