AMD · 2 hours ago
Sr. PyTorch Software Development Engineer
Austin, TX
Full-time
Onsite
Senior Level
$164K/yr - $246K/yrAMD is a leading company focused on building innovative products that enhance computing experiences across various domains, including AI and data centers. They are seeking a Senior PyTorch Software Development Engineer to drive the development and optimization of Deep Learning models on AMD GPUs and enhance the PyTorch ecosystem. The role involves collaboration with internal teams and the open-source community to deliver high-performance solutions for AI workloads.
Artificial Intelligence (AI)Cloud ComputingComputerEmbedded SystemsGPUHardwareSemiconductor
Responsibilities
Model Development and Optimization: Drive the development and optimization of Deep Learning models on AMD GPUs, ensuring high performance and scalability across training and inference workloads
Framework Enablement: Lead efforts to enhance PyTorch and its ecosystems on AMD GPUs, contributing performance-critical improvements to upstream open-source repositories
Open-Source Integration: Engage directly with PyTorch and other open-source maintainers to align on requirements, propose enhancements and ensure AMD contributions are successfully adopted and maintained upstream
Community Feature Enablement: Collaborate with PyTorch community and ecosystem partners to design, enable and upstream new features, ensuring AMD GPUs remain first class for emerging AI workloads
Cross-Functional Collaboration: Partner closely with internal GPU Libraries, compilers, and runtime teams to identify performance bottlenecks and deliver optimized solutions for large-scale AI workloads
Scalable Systems Optimization: Developed innovative solutions for both scale-up (multi-GPU) and scale-out (multi-node) distributed training environments to optimize performance on AMD GPUs
Engineering Excellence: Apply rigorous software engineering best practices, including code review, testing, CI/CD integration and debugging, to deliver reliable and maintainable solutions at scale
Qualification
Required
Strong technical foundation in C++/Python AI development in Linux environment
Deep technical expertise with strong analytical abilities
Ability to excel both as a collaborative team member and as an independent contributor capable of defining goals, scoping work and leading development efforts end to end
Drive the development and optimization of Deep Learning models on AMD GPUs, ensuring high performance and scalability across training and inference workloads
Lead efforts to enhance PyTorch and its ecosystems on AMD GPUs, contributing performance-critical improvements to upstream open-source repositories
Engage directly with PyTorch and other open-source maintainers to align on requirements, propose enhancements and ensure AMD contributions are successfully adopted and maintained upstream
Collaborate with PyTorch community and ecosystem partners to design, enable and upstream new features, ensuring AMD GPUs remain first class for emerging AI workloads
Partner closely with internal GPU Libraries, compilers, and runtime teams to identify performance bottlenecks and deliver optimized solutions for large-scale AI workloads
Develop innovative solutions for both scale-up (multi-GPU) and scale-out (multi-node) distributed training environments to optimize performance on AMD GPUs
Apply rigorous software engineering best practices, including code review, testing, CI/CD integration and debugging, to deliver reliable and maintainable solutions at scale
Master's or PhD level degree in Computer Science, Computer Engineering, Electrical Engineering, or a related field
Preferred
Expert skills in Python and C++, with experience in debugging, performance tuning, and test design to ensure high-quality, maintainable software solutions
Solid understanding of AI frameworks such as PyTorch, Triton and vLLM, with applied knowledge across domains such as Natural Language Processing, Vision, Audio and Recommendation Systems
Expert experience with GPU Programming models (CUDA, HIP)
Proficiency with modern software practices and tooling, including debuggers, profilers, and version control systems (e.g., GitHub)
Effective communicator with strong analytical and problem-solving skills, capable of working across diverse technical teams
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