Sr AI Software Development Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

AMD · 17 hours ago

Sr AI Software Development Engineer

AMD is a company focused on building products that accelerate next-generation computing experiences, including AI and data centers. They are seeking a talented AI Software Development Engineer to join their team, working on software for GPU computational accelerators and optimizing performance for AI applications.

AI InfrastructureArtificial Intelligence (AI)Cloud ComputingComputerEmbedded SystemsGPUHardwareSemiconductor
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

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
Work with a team of Software Engineers to push the boundaries of efficiency and performance

Qualification

C++PythonDeep Learning frameworksGPU performance optimizationOpen-source software developmentPerformance analysisAnalytical skillsWillingness to learnPublications in ML conferencesProfiling tools expertiseParallel methods optimizationContainerization toolsProblem-solving skillsTeam collaborationCommunication skills

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 or PhD 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

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

Advanced Micro Devices is a semiconductor company that designs and develops graphics units, processors, and media solutions.

H1B Sponsorship

AMD has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (836)
2024 (770)
2023 (551)
2022 (739)
2021 (519)
2020 (547)

Funding

Current Stage
Public Company
Total Funding
unknown
Key Investors
OpenAIDaniel Loeb
2025-10-06Post Ipo Equity
2023-03-02Post Ipo Equity
2021-06-29Post Ipo Equity

Leadership Team

leader-logo
Lisa Su
Chair & CEO
linkedin
leader-logo
Mark Papermaster
CTO and EVP
linkedin
Company data provided by crunchbase