Senior Staff Software Engineer - AI Frameworks / Compiler / Systems
Advanced Micro Devices
- San Jose, CA
- Permanent
- Full-time
- Experience with development in one of the focus areas: AI frameworks, AI compiler stack, and/or performance tuning and optimizations for workloads running on ML accelerator HWs.
- Experience with ML frameworks such as PyTorch, OnnxRuntime, JAX, TensorFlow
- Proficient in C++ programming.
- Experience developing and debugging in Python.
- Excellent skills at designing Python tools and libraries used by large number of users.
- Experience with AI model architectures, e.g. Transformers, CNNs.
- Team player and ready to work with a geographically distributed team.
- Highly preferred - knowledge of custom accelerator hardware
- Depth of experience with AI compilers, dataflow graphs, MLIR, and dialects, compiler optimizations, hardware specific optimizations.
- Experience with MLIR, LLVM, IREE, XLA, TVM, Halide is a plus.
- OR Depth of experience with model optimization, including quantization, sparsity.
- Solid understanding of transformer architecture, LLMs, MoE.
- Experience on LLMs/LMMs finetuning, distillation, and/or RLHF