AMD模型优化工程师（推理&训练）Model Optimization Engineer (Inference & Training)

工作描述

任职要求
* Strong software engineering in Python and C/C++. * Practical experience with PyTorch/JAX and building/extending deep learning frameworks. * Hands‑on CUDA and/or ROCm development; experience writing or optimizing GPU kernels. * Experience with Triton (kernel development/optimization) is highly desired. * Proven experience with model optimization techniques, especially low‑bitwidth quantization and other compression methods. * Familiarity with GenAI inference engines and optimizations (e.g., vLLM, SGLang, xDiT, continuous batching, speculative decoding). * Skilled at profiling and performance debugging across stack layers (operator → model → framework → hardware). PREFERRED QUALIFICATIONS * Publications or contributions in model optimization / ML systems are a strong plus. * Experience with distributed training/inference frameworks (e.g., Torchtitan, Megatron, DeepSpeed, HF Accelerate, vLLM, SGLang, xDiT). * Background in PTQ/QAT quantization algorithms, efficient attention variants, or low-bitwidth/sparse kernels. * Familiarity with rea…