AMD · 6 days ago
Sr. Staff Engineer, AI Models and Applications
San Jose, CA
Full-time
Hybrid
Senior Level, Lead/Staff
$192K/yr - $288K/yrAMD is a company dedicated to building innovative products that enhance computing experiences across various domains. They are seeking a specialized Sr. Staff Engineer to enable efficient Generative AI training and inference at scale, working with a team of talented specialists and cutting-edge technology.
AI InfrastructureArtificial Intelligence (AI)Cloud ComputingComputerEmbedded SystemsGPUHardwareSemiconductor
Responsibilities
Propose and apply innovative techniques to support both training and inferencing including innovative transformer architectures, parallelism strategies to train on large clusters, low-precision training
Implement novel efficient architectures for Generative AI models for training and inference and showcase benefits on AMD
Work with open-source framework and community (e.g., PyTorch, JAX, Hugging Face) to integrate AMD optimized models, libraries and publish training recipes
Collaborate with software and hardware team to E2E co-optimize performance on current and future AMD solutions
Publish and promote your work within AMD and at external venues
Qualification
Required
Deep technical understanding of the latest generative AI applications like large language models (LLMs), large multimodal models (LMMs), image/video generation
Experience training models at scale
Passionate about innovating efficient approaches to enable distributed training and inference at scale on AMD devices
Propose and apply innovative techniques to support both training and inferencing including innovative transformer architectures, parallelism strategies to train on large clusters, low-precision training
Implement novel efficient architectures for Generative AI models for training and inference and showcase benefits on AMD
Work with open-source framework and community (e.g., PyTorch, JAX, Hugging Face) to integrate AMD optimized models, libraries and publish training recipes
Collaborate with software and hardware team to E2E co-optimize performance on current and future AMD solutions
Publish and promote your work within AMD and at external venues
PhD or master's degree or above; with major in CS, EE, Mathematics, or a related field
Preferred
Strong technical expertise in Gen AI model training and inference, and familiarity working with deep learning frameworks like PyTorch/JAX
Strong technical expertise in algorithmic innovation towards efficient Gen AI application for both training and inferencing
Expertise/publications in one of the areas preferred - efficient model architectures, optimized training, innovative parallelism strategies or low-precision training
Additional plus if publications include conferences such as NeuRIPS, CVPR, ECCV/ICCV, ICML, ICLR, etc
Experience productizing generative AI models and training foundation model at Scale
Excellent written, verbal, and presentation skills, ability to coordinate internally and externally
Several years of experience in AI, deep learning and related software development
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
Italian Startups - Startupbusiness.it
2026-01-07
2026-01-07
Company data provided by crunchbase