- Shanghai, Shanghai, China
Optimize and develop deep learning frameworks for AMD GPUs, enhancing performance across multi-GPU and multi-node systems.
Responsibilities
- Optimize major DL/LLM frameworks (TensorFlow, PyTorch, vLLM, SGLang) for AMD GPUs
- Develop and tune GPU kernels and performance-critical operators
- Adapt and optimize LLM architectures (Llama, Qwen, DeepSeek)
- Perform profiling to identify bottlenecks and implement system, memory, and communication optimizations
- Leverage advanced compiler technologies and graph compilers
Requirements
- Bachelor’s or Master’s Degree in Computer Science, Computer Engineering, Electrical Engineering or related field
- Nice to have: CUDA C++
- Nice to have: HIP
- Nice to have: ASM
- Nice to have: CK