Senior Fellow, ML Optimization jobs in United States
cer-icon
Apply on Employer Site
company-logo

AMD · 1 week ago

Senior Fellow, ML Optimization

AMD is a company focused on building innovative products that enhance computing experiences across various domains. The Senior Fellow role is a technical leadership position responsible for defining and driving strategies for ML workload performance optimization on AMD GPUs, requiring collaboration across multiple teams and expertise in AI frameworks and performance analysis.

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

Responsibilities

Set strategy and roadmap for AMD model optimization
Performance tuning, profiling and analysis of large-scale models for LLM, diffusion, multimodal, RecSys and generative AI, single node and distributed. In addition to exploring various tradeoffs and design decisions
Participate in hardware-software co-design for future hardware optimization on various ML workloads
Develop and improve framework, tools and infrastructure for performance estimation, modeling and reporting
Communicate and present the results of the performance analysis and modeling to stakeholders, and senior leadership. And provide a concrete recommendation
Cross team collaboration and working across the organization to identify opportunities and develop strategies

Qualification

ML hardware architectureSoftware optimizationPerformance modelingAI frameworksPerformance analysisTechnical leadershipCollaborationCommunication skillsMentoring

Required

Deep knowledge with ML hardware architecture, software optimization, performance modeling, AI frameworks and latest trend in inference and training optimization
Hand-on experience in mapping model architecture to low level software, hardware and understanding the impact of each layer of the stack on model performance
Strong knowledge in latest generative model architecture, especially SoTA models, distributed inference and deployment at scale
Set strategy and roadmap for AMD model optimization
Performance tuning, profiling and analysis of large-scale models for LLM, diffusion, multimodal, RecSys and generative AI, single node and distributed
Participate in hardware-software co-design for future hardware optimization on various ML workloads
Develop and improve framework, tools and infrastructure for performance estimation, modeling and reporting
Communicate and present the results of the performance analysis and modeling to stakeholders, and senior leadership
Cross team collaboration and working across the organization to identify opportunities and develop strategies
A PhD or Master plus equivalent experience in computer science, electrical engineer, or a related field

Preferred

Multiple years of technical experience in performance optimization
Strong technical expertise and experience in performance analysis, projection, and hardware architecture
Deep knowledge and hand-on experience of AI Frameworks such as PyTorch, JAX, vLLM, and SGLang
Strong technical leadership skills, ability to work collaboratively with cross-functional teams
Mentor, coach, and inspire a diverse and talented team of researchers and engineers
Excellent written, verbal, and presentation skills, ability to coordinate internally and externally

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