AMD GPU Performance Engineer
We're looking for an experienced GPU Performance Engineer to help make vLLM a high-performance inference engine across AMD GPUs. You'll work on cutting-edge AI infrastructure, optimizing GPU kernels and inference performance using AMD's ROCm ecosystem.
Key Skills:
✅ AMD GPU optimization: ROCm, HIP, Triton, CK, AITER✅ GPU kernel development and performance tuning✅ LLM inference: vLLM, SGLang, or TensorRT-LLM✅ Profiling, benchmarking, and performance optimization✅ GPU architecture, memory optimization, and kernel execution✅ Experience with attention, GEMM, KV cache, or fused kernels Open-source contributions to vLLM, ROCm, Triton, or PyTorch are a plus. This is an opportunity to solve challenging GPU performance problems and contribute to open-source AI infrastructure used across the ecosystem.