Software Engineer III : Ai/ml Solutions

JPMorgan Chase JPMorgan Chase · Banking · Palo Alto, CA +1 · Consumer & Community Banking

Software Engineer III (AI/Machine Learning Platform Engineer) at JPMorgan Chase, responsible for building, scaling, and maintaining robust machine learning platforms and infrastructure. The role involves designing and optimizing tools for the end-to-end ML lifecycle, including training, deployment, and monitoring, and integrating various ML capabilities into unified platform solutions. Focus is on implementing secure production code, collaborating with data scientists and ML engineers, and ensuring platform reliability and performance.

What you'd actually do

  1. Design, build, and maintain scalable machine learning platforms and infrastructure to support end-to-end ML workflows.
  2. Develop and optimize tools for model training, deployment, monitoring, and lifecycle management.
  3. Integrate data engineering, feature management, and model serving capabilities into unified ML platform solutions.
  4. Implement secure, high-quality production code for platform services, APIs, and automation pipelines.
  5. Collaborate with data scientists, ML engineers, and product teams to understand requirements and deliver platform features that accelerate ML development and operations.

Skills

Required

  • Software engineering concepts
  • Python
  • ML frameworks (TensorFlow, PyTorch, Scikit-learn)
  • Data processing frameworks (Spark, Pandas, SQL)
  • Cloud-based ML platforms (AWS SageMaker, GCP AI Platform, Azure ML) or on-prem ML infrastructure
  • MLOps practices (CI/CD for ML, model versioning, monitoring)
  • API development
  • Platform services development
  • Software development life cycle
  • Agile methodologies

Nice to have

  • Databricks
  • Snowflake
  • Snorkel AI
  • Docker
  • Kubernetes
  • Airflow
  • Feature stores
  • Model registries
  • ML metadata management
  • Infrastructure-as-code tools (Terraform, CloudFormation)
  • RESTful APIs
  • Microservices architectures

What the JD emphasized

  • Hands-on experience building, deploying, and maintaining machine learning platforms or infrastructure
  • Proficiency 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

Other signals

  • building scalable ML platforms
  • deploying and monitoring models
  • MLOps practices
  • cloud-based ML platforms