12 hours ago
Pudong, ChinaSenior
Responsibilities
- Design, develop, and optimize high-performance AI operators and GPU kernels for training and inference workloads.
- Analyze and improve kernel performance through profiling, memory optimization, kernel fusion, parallelization, and architecture-aware tuning.
- Develop GPU kernels using CUDA, HIP, Triton, or similar frameworks.
- Build AI-powered coding agents and automation workflows for kernel generation, optimization, benchmarking, and validation.
- Collaborate with model and infrastructure teams to improve end-to-end AI system performance.
Requirements
- Bachelor's degree or above in Computer Science, Computer Engineering, Artificial Intelligence, or a related field.
- At least 4 years of relevant experience.
- Strong programming skills in C++ and/or Python.
- Hands-on experience developing and optimizing GPU kernels using CUDA, HIP, Triton, or similar frameworks.
- Understanding of GPU architecture and performance optimization, including memory hierarchy, occupancy, register pressure, and shared memory.
- Experience with profiling and optimization tools such as Nsight or rocprof.
- Experience with performance-critical AI operators such as GEMM, Attention, MoE, or Embedding.
- Preferred: experience with kernel fusion, performance modeling, communication/computation overlap, coding agents, code generation systems, AI-assisted software engineering tools, or large-scale AI training and inference systems.
- Nice to have: experience with LLM post-training techniques such as SFT, RLHF, DPO, or GRPO, plus publications or open-source contributions in GPU optimization, AI systems, or coding agents.
Categories
About AMD
AMD designs and sells CPUs, GPUs, and adaptive/embedded computing products for PCs, data centers, gaming, and edge devices. Its portfolio includes Ryzen and EPYC processors, Radeon and Instinct graphics, and adaptive SoCs from its Xilinx acquisition, sold to OEMs, cloud providers, and device makers. Founded in 1969 and headquartered in Santa Clara, it is a public company on NASDAQ and supplies semi-custom chips for major game consoles.