Staff AI Research Engineer

AMD AMD · Semiconductors · Santa Clara, CA · Engineering

AMD is seeking a Staff AI Research Engineer to join their Llama team. The role involves researching, prototyping, and implementing cutting-edge generative AI techniques, including post-training reasoning and RL methods. A key responsibility is translating these advanced AI concepts into clear educational content for a broad developer audience, leveraging AMD's high-performance chips. The position also requires staying current with AI research and contributing to internal knowledge sharing, with potential development in GPU kernel writing and DSLs.

What you'd actually do

  1. Research, prototype, and implement cutting-edge techniques in generative AI, e.g. reasoning post-training, the latest RL methods, etc.
  2. Translate advanced AI concepts into clear, engaging educational content for a broad audience
  3. Stay current with the latest research in generative models and contribute to internal knowledge sharing
  4. Develop skills in GPU kernel writing, triton, or DSLs.

Skills

Required

  • Python
  • PyTorch
  • TensorFlow
  • JAX
  • LLMs/transformer architectures

Nice to have

  • Experience working in cross-functional teams
  • Contributions to open-source projects
  • research publications
  • internal technical blogs
  • simplify and communicate complex technical concepts
  • GPU kernel writing
  • triton
  • DSLs

What the JD emphasized

  • implementing the latest AI research
  • reasoning post-training
  • latest RL methods

Other signals

  • train language models
  • implementing the latest AI research
  • reasoning post-training
  • latest RL methods