Hw/sw Co-design Engineer

AMD AMD · Semiconductors · MARKHAM, Canada · Engineering

AMD is seeking a HW/SW Co-design Engineer to collaborate on next-generation GPU solutions for data centers and super-computers. The role involves architecture exploration, kernel prototyping, modeling, and analysis of ML/HPC workloads, with a focus on optimizing GPU kernels and building libraries for key ML operators.

What you'd actually do

  1. Write and optimize GPU kernels at different levels of abstractions (assembly, Hip, Triton, .. etc).
  2. Write microbenchmarks that can be used for competitive analysis and performance verification.
  3. Prototype ideas across the stack from software, and firmware to architectural models to evaluate and validate ideas.
  4. Collect/summarize data and simulation results for consumption by architects and design teams
  5. Work with architects to propose innovative solutions that can be implemented in next generation HW and SW.

Skills

Required

  • C
  • C++
  • Python

Nice to have

  • GPU architectures
  • CPU architecture
  • machine learning (ML) networks
  • TensorFlow
  • PyTorch
  • building libraries for key ML operators
  • GEMMs
  • Attention
  • Collectives
  • Graphics and Compute API's
  • CUDA
  • OpenCL
  • Vulkan
  • compiler stack
  • graph compilers
  • MLIR
  • LLVM
  • Network-on-Chip (NoC) design
  • interconnect systems
  • operating systems (OS)
  • device driver development
  • hardware modeling
  • design using RTL
  • SystemC

What the JD emphasized

  • ML/HPC workloads
  • GPU kernels
  • ML operators

Other signals

  • ML/HPC workloads
  • GPU kernels
  • ML operators