The AI Models and Applications team at AMD is seeking a specialized Sr. Staff level engineer focused on enabling innovative and efficient Generative AI training and inference at scale. This role involves working with a talented team to scale training and inference for the latest Generative AI models. The ideal candidate will have a deep technical understanding of generative AI applications and experience in training models at scale. The position offers opportunities to influence AI model training and inference optimizations across various applications.

• PhD or master's degree in Computer Science, Electrical Engineering, Mathematics, or a related field.
• Strong technical expertise in Generative AI model training and inference.
• Familiarity with deep learning frameworks such as PyTorch and JAX.
• Several years of experience in AI, deep learning, and related software development.
• Excellent written, verbal, and presentation skills.

• Propose and apply innovative techniques to support both training and inference.
• Implement novel efficient architectures for Generative AI models for training and inference.
• Work with open-source frameworks and communities to integrate AMD optimized models and libraries.
• Collaborate with software and hardware teams to co-optimize performance on AMD solutions.
• Publish and promote work within AMD and at external venues.

• AMD provides a competitive 'Total Rewards' package that focuses on financial growth, health, and work-life balance.