AI Software Product Engineer(GPU Kernel)
Advanced Micro Devices View all jobs
- Shanghai
- Permanent
- Full-time
- Hands-on experience with AI frameworks, including PyTorch, vLLM, SGLang, Unsloth, TensorRT-LLM, Megatron-LM, and DeepSpeed.
- Proven experience in LLMs, Generative AI models, transformer architectures, and end-to-end AI pipelines.
- Familiarity with AMD MI-series GPU architecture, GPU kernel programming, and the ROCm AI software stack is strongly preferred.
- Strong communication and presentation skills, with the ability to articulate architectural proposals and value propositions clearly.
- BS required; MS preferred, with 6+ years of relevant industry experience.
- Lead and contribute to AI open-source software projects that support the developer community and the broader ecosystem.
- Drive developer enablement through technical content (blogs, tutorials, user guides) and AI Academy initiatives.
- Support the success of AI developers, communities, and customer PoCs through hands-on technical contributions.
- Capture and prioritize developer and customer requirements to influence AMD’s AI software and solutions roadmap.
- Analyze competitive AI software and solutions to identify strengths/weaknesses and clearly communicate AMD’s value propositions.
- Provide feedback and requirements for AI software across cloud, client, and edge deployments.