AMD · 18 hours ago
AI Models GPU deployment software Engineer
Austin, TX
Full-time
Onsite
Mid Level
$143K/yr - $215K/yrAMD is a company dedicated to building innovative products that enhance computing experiences across various domains, including AI and data centers. The role involves optimizing AI models and frameworks on AMD GPUs, collaborating with internal teams, and applying best software engineering practices.
Artificial Intelligence (AI)Cloud ComputingComputerEmbedded SystemsGPUHardwareSemiconductor
Responsibilities
Enable and optimize key AI models (LLM, Vision, MultiModal, etc.) on AMD GPUs
Optimize AI frameworks like PyTorch, TensorFlow, etc. on AMD GPUs in upstream open-source repositories
Collaborate and interact with internal GPU library teams to analyze and optimize training and inference for AI
Work with open-source framework maintainers to understand their requirements - and have your code changes integrated upstream
Optimize GPU kernels, understand and drive AI operator performance (GEMM, Attention, etc.) with specialized teams
Work in a distributed computing setting to optimize for both scale-up (multi-GPU) and scale-out (multi-node) systems
Apply your knowledge of software engineering best practices
Qualification
Required
Strong technical and analytical skills in C++/Python AI development in a Linux environment
Ability to work as part of a team, while also being able to work independently, define goals and scope and lead your own development effort
Enable and optimize key AI models (LLM, Vision, MultiModal, etc.) on AMD GPUs
Optimize AI frameworks like PyTorch, TensorFlow, etc. on AMD GPUs in upstream open-source repositories
Collaborate and interact with internal GPU library teams to analyze and optimize training and inference for AI
Work with open-source framework maintainers to understand their requirements - and have your code changes integrated upstream
Optimize GPU kernels, understand and drive AI operator performance (GEMM, Attention, etc.) with specialized teams
Work in a distributed computing setting to optimize for both scale-up (multi-GPU) and scale-out (multi-node) systems
Apply your knowledge of software engineering best practices
Masters or PhD or equivalent experience in Computer Science, Computer Engineering, or related field
Preferred
Knowledge of GPU computing (HIP, CUDA, OpenCL)
AI model experience or knowledge - Natural Language Processing, Vision, Audio, Recommendation systems
Excellent C/C++/Python programming and software design skills, including debugging, performance analysis, and test design
Experiences to run workloads on large scale heterogeneous cluster is a plus
Experiences to optimize GPU kernels for performance is a plus
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.
Founded in 1969
10001+ employees
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 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
2025-12-30
2025-12-28
Investing.com
2025-12-27
Company data provided by crunchbase