Senior ML Platform Engineer (GPU/Cloud)

JPMorgan Chase; Palo Alto (CA); Vor Ort; USD 180.000 - 250.000; Vollzeit

Zusammenfassung

JPMorganChase in Palo Alto is seeking a Software Engineer III to join the AI/ML data platform team. You will design and operate scalable ML training systems on AWS and other clouds, productionizing training workloads, and improving performance and cost efficiency.

You will work on end-to-end pipelines, Kubernetes, EKS, AI tooling, and governance. The role requires strong Python, ML frameworks, and cloud experience with a focus on reliability and secure workflows.

Qualifikationen

• Formal training or certification on software engineering concepts and 3+ years applied experience.
• Demonstrated experience running ML training in cloud environments and debugging issues across infrastructure & code.
• Strong Python skills with solid engineering practices (testing, code reviews, modular design, dependency management).
• Experience building automation/CI for ML codebases (build, test, release, deployment/promotion workflows).
• Hands on experience with deep learning training workflows and at least one major framework (eg., PyTorch or TensorFlow).
• Understanding of training performance and stability: data loading bottlenecks, mixed precision, checkpointing, reproducibility, and evaluation methodology.
• Experience with distributed training and related concepts (e.g., DDP/FSDP/DeepSpeed concepts, collective communication basics, scaling and bottleneck analysis).
• Ability to profile and optimize training systems (CPU/GPU utilization, memory, I/O throughput, networking, scheduling).
• Experience with Kubernetes fundamentals for running compute-intensive workloads and AWS (eg., EKS/ECR, S3, IAM, VPC/networking, Cloudwatch, EC2)
• Hands-on experience using enterprise-authorized AI-assisted software development tools within the work environment with demonstrated ability to evaluate and refine outputs for correctness, performance, and security.
• Understanding of responsible AI use in engineering workflows, including data sensitivity considerations, secure handling of inputs/outputs, and adherence to resiliency and security expectations.

Aufgaben

• Design, build, and maintain end-to-end ML training platform.
• Run and optimize GPU training workloads (single-node and distributed), improving throughput, utilization and reproducibility.
• Build and operate training infrastructure on Kubernetes (e.g., EKS and other manage Kubernetes platforms), including resource management and workload troubleshooting.
• Enable Gen AI/LLM training and fine-tuning workflows, including evaluation harnesses, artifact/version governance, and scalable GPU execution patterns aligned to enterprise controls.
• Implement observability for training systems: metrics, logs, dashboards, alerting, and operational runbooks.
• Partner with data engineering and platform teams to define interfaces, standards, and guardrails (security, access, cost controls).
• Improve developer experience for training: standardized containers, CI/CD, templates, documentation, and self-service workflow.
• Leverages enterprise-authorized AI coding assist tools to improve code quality, delivery speed, and productivity; validates outputs through peer review and testing.

Tools

Kubernetes

AWS

EKS

S3

IAM

VPC

CloudWatch

EC2

Jobbeschreibung

JPMorganChase in Palo Alto is seeking a Software Engineer III to join the AI/ML data platform team. You will design and operate scalable ML training systems on AWS and other clouds, productionizing training workloads, and improving performance and cost efficiency.

You will work on end-to-end pipelines, Kubernetes, EKS, AI tooling, and governance. The role requires strong Python, ML frameworks, and cloud experience with a focus on reliability and secure workflows.