AMD · 2 weeks ago
Fellow, AI Performance Software Engineer
California, United States
Full-time
Onsite
Mid, Senior Level
$223K/yr - $334K/yr
4+ years expAMD is a leader in next-generation computing experiences, focusing on AI and data centers. They are seeking a highly motivated AI Software Engineer to optimize software for GPU computational accelerators and enable deep learning models in cloud and on-prem environments.
AI InfrastructureArtificial Intelligence (AI)Cloud ComputingComputerEmbedded SystemsGPUHardwareSemiconductor
No H1B
Responsibilities
Work with a team of Software Engineers to enable DL models, libraries, and applications for Instinct GPUs in both on-prem and Cloud environments
Analyze and optimize the performance of AI software and understand hardware bottlenecks and harness performance to hit close to roofline
Develop software and system-level performance optimizations with a solid architecture understanding in GPUs
Collaborate with community maintainers and submit contributions in open-source software development
Provide clear and timely communication related to status and other key aspects of the project to the leadership team
Qualification
Required
Strong programming skills in C++ and Python
Strong development experience is at least one major DL framework such as Pytorch or Tensorflow in inference, fine tuning and/or training
MS with years of related experience or PhD with years of related experience in Computer Science or Computer Engineering or related equivalent
Experience developing software and system-level performance optimizations with a solid architecture understanding in GPUs a plus
Experience with open-source software development including collaboration with community maintainers and submitting contributions is a plus
Publications in reputed peer-reviewed ML conferences/journals a plus
Excellent analytical and problem-solving skills root-causing/addressing performance issues
Ability to work independently and as part of a team
Willingness to learn skills, tools, and methods to advance the quality, consistency, and timeliness of AMD software products
Minimum 4 years of experience required
Preferred
Expertise in profiling tools across the AI SW Stack (Torchprofiler, RocM profiler, Vtune, Nsight)
Experience in implementing and optimizing parallel methods on GPU accelerators (NCCL/RCCL, OpenMP, MPI)
Performance analysis skills for both CPU and GPU
Experience with Singularity, Docker, and/or Kubernetes
Experience providing clear and timely communication related to status and other key aspects of the project to leadership team
Benefits
AMD benefits at a glance.
Company
AMD
Advanced Micro Devices is a semiconductor company that designs and develops graphics units, processors, and media solutions.
10001+ employees
Funding
Current Stage
Public CompanyTotal Funding
unknownKey Investors
OpenAIDaniel Loeb
2025-10-06Post Ipo Equity
2023-03-02Post Ipo Equity
2021-06-29Post Ipo Equity
Leadership Team
Lisa Su
Chair & CEO
Mark Papermaster
CTO and EVP
Recent News
2026-02-06
The Next Platform
2026-02-06
2026-02-06
Company data provided by crunchbase