AI Systems Engineer - HPC

Advanced Micro Devices, Incabout 1 year ago
San Jose, California, United States
Hybrid
Full-time
Junior Level (1-3 years)

Job Description

Position Overview

At AMD, we care deeply about transforming lives with our technology. Our mission is to build great products that accelerate next-generation computing experiences across data centers, artificial intelligence, PCs, gaming, and embedded systems. We push the limits of innovation to solve the world’s most important challenges while embracing a culture of execution excellence, directness, humility, collaboration, and inclusivity.

As an AI Systems Engineer within the AMD IT Compute Platforms Engineering team, you will design, develop, and administer High-Performance Computing (HPC) infrastructure, GPU clusters, and AI workload schedulers. You are passionate about large-scale distributed computing and continuously seek to build scalable, high-performance HPC/AI/Data services by leveraging AMD hardware, software, and innovative processes.

Key Responsibilities

  • Develop, implement, and maintain GPU-based clusters to ensure optimal performance
  • Administer ML/AI platforms – including distributed ML services, LLMs, and AI inferencing – by managing deployments, resource allocation, monitoring, and security
  • Automate system provisioning and cluster management end to end
  • Collaborate with cross-functional teams to address AI infrastructure requirements and support AI-related projects
  • Monitor and evaluate the performance of AI systems and clusters to ensure adherence to industry best practices and company standards
  • Utilize AI/ML to continuously improve internal processes and tools for end-to-end service delivery

Preferred Qualifications

  • Experience in developing Python-based AI applications and user interfaces
  • Expertise in HPC infrastructure engineering for AI/HPC workloads
  • Proficiency with SLURM and Kubernetes management
  • Experience managing GPU clusters to optimize GPU-based services, tools, and software
  • Background in creating web services with an HPC backend (such as AI)
  • Familiarity with RoCEv2, K8s, KVM, Ubuntu, Python, Shell, GPU drivers, and 400G networking for cluster interconnects
  • Demonstrated experience with AI workload schedulers and allocation optimization
  • Experience with automation and monitoring tools such as Ansible, Saltstack, Terraform, Prometheus, and Grafana
  • Strong organizational, problem-solving, and troubleshooting skills, coupled with excellent verbal and written communication abilities

Benefits & Perks

  • AMD benefits at a glance

Required Skills

GPU cluster management
SLURM
HPC infrastructure engineering
Monitoring tools (Prometheus, Grafana)
AI/ML application development
Linux (Ubuntu)
Automation (Ansible, Saltstack, Terraform)
Shell scripting
Kubernetes
Distributed computing
Python
System provisioning
AI workload scheduling
Networking (RoCEv2, 400G interconnect)
KVM