Staff AI Research Engineer

The Role

We are hiring a highly technical and creative AI Research Engineer to join our AI team. You will research, prototype, and implement advanced AI techniques, contribute to internal innovation, and help share practical AI learnings for developers building on high-performance AMD hardware.

THE PERSON

You are comfortable turning cutting-edge research into practical outcomes. You can explain complex AI ideas clearly and build prototypes that demonstrate real-world impact. You stay current with fast-moving developments in generative AI and enjoy collaborating across teams. You are motivated by problems with substantial headroom for improvement, where thoughtful engineering and research can deliver outsized impact.

Key Responsibilities

• Research, prototype, and implement modern AI methods (for example: reasoning post-training, reinforcement learning approaches, and new generative model techniques).
• Translate advanced AI concepts into clear technical content and demos for broad developer audiences.
• Contribute to internal knowledge sharing around model training, inference, and optimization.
• Build and improve practical workflows for deploying and showcasing AI capabilities on AMD platforms.
• Drive code optimization, debugging, and root-cause analysis across research prototypes and production-oriented AI workflows.

Preferred Experience

• Experience working in cross-functional teams in fast-paced technical environments.
• Publications at top-tier venues (for example: NeurIPS, ICLR, ICML, CVPR, ECCV, ICCV, COLM, or similar).
• Contributions to open-source projects or internal/external engineering blogs.
• Strong proficiency in Python and deep learning frameworks such as PyTorch, TensorFlow, or JAX, plus effective use of GenAI tools to accelerate coding, prototyping, demos, and implementation.
• Demonstrated experience training and scaling LLMs and transformer-based models, including challenging training tasks, distributed/parallel training, and performance tuning across large workloads.
• Proven ability to simplify and communicate complex technical ideas through presentations, demos, and written content.

Academic Credentials

• Bachelor's degree in Computer Science, Computer Engineering, Electrical Engineering, or related field.
• Master's degree preferred; PhD is a strong plus, especially with publications in distributed systems, AI infrastructure, or GPU computing.

Benefits

Benefits offered are described: AMD benefits at a glance.