Advanced Micro Devices-posted 3 months ago
Full-time • Mid Level
Hybrid • San Jose, CA
5,001-10,000 employees
Computer and Electronic Product Manufacturing

AMD is looking for a world class AI frameworks engineer who can provide technical leadership in the development of various AI frameworks in the AMD ecosystem. You will need to drive technical direction for next generation frameworks for AI model training and inference for wide variety of AMD devices, current and future, such as MI Instinct, and Radeon GPUs, XDNA devices, including the recently released Ryzen AI, Alveo V70 and Versal ACAP, and datacenter CPUs such as EPYC. You will work enhance the AI framework capabilities to enable cutting-edge models on onto AMD's cutting-edge hardware.

  • Drive technical direction for next generation frameworks for AI model training and inference.
  • Enhance AI framework capabilities to enable cutting-edge models on AMD hardware.
  • Collaborate closely with AI researchers to develop framework components for efficient AI model mapping.
  • Guide other senior developers and domain experts in next generation framework software.
  • Experience with development in AI frameworks, AI runtime stacks, and/or performance tuning for ML accelerator HWs.
  • Experience with ML frameworks such as PyTorch, OnnxRuntime, JAX, TensorFlow.
  • Proficient in C++ programming.
  • Experience developing and debugging in Python.
  • Experience with AI model architectures, e.g. Transformers, CNNs.
  • Team player ready to work with a geographically distributed team.
  • Knowledge of custom accelerator hardware.
  • Experience with AI software framework, benchmarking and profiling.
  • Operator/kernel development and deployment of models to accelerator (GPU/NPU) stack.
  • Depth of experience with model optimization, including quantization, sparsity.
  • Understanding of model architectures, LLMs, MoE, diffusion.
  • AMD benefits at a glance.
© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service