Hw/sw Co-design Engineer

AMD AMD · Semiconductors · MARKHAM, Canada · Engineering

This role focuses on the hardware/software co-design of next-generation GPU solutions, specifically for ML/HPC workloads. The engineer will optimize GPU kernels, prototype ideas across the stack, and work with architects to propose innovative solutions. Experience with ML networks, building libraries for ML operators, and compiler stacks is preferred.

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
  • ML operators
  • CUDA
  • OpenCL
  • Vulkan
  • compiler stack
  • MLIR
  • LLVM
  • Network-on-Chip (NoC) design
  • interconnect systems
  • operating systems (OS)
  • device driver development
  • hardware modeling
  • RTL
  • SystemC

What the JD emphasized

  • ML/HPC workloads
  • GPU kernels
  • ML operators

Other signals

  • ML/HPC workloads
  • GPU kernels
  • ML operators