The AI Research Engineer role at AMD focuses on advancing generative AI through research and prototyping. The position is part of the Llama team, which is dedicated to making AI knowledge accessible to developers. The ideal candidate will be highly technical and creative, contributing to the training of language models and the creation of educational content. This role is essential for scaling AI education efforts and driving innovation within the team.

• Bachelor's degree in Computer Science, Computer Engineering, Electrical Engineering, or a related field.
• Proficiency in Python and deep learning frameworks such as PyTorch, TensorFlow, or JAX.
• Demonstrated experience with LLMs and transformer architectures.
• Experience working in cross-functional teams, especially in fast-paced environments.

• Research, prototype, and implement cutting-edge techniques in generative AI.
• Translate advanced AI concepts into clear, engaging educational content for a broad audience.
• Stay current with the latest research in generative models and contribute to internal knowledge sharing.
• Develop skills in GPU kernel writing, triton, or DSLs.

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