Senior GPU Kernel Optimization Engineer

AMD
Beijing
Workplace: OnsiteFull timeCNY 411,670 - 588,100 annuallyFunction: Communications, PR & CommunityExperience: 4+ yearsEducation: bachelorsSkills: ["Communication","Collaboration","Problem-solving"]

Join AMD as an LLM Training & Agent Engineer to train, fine-tune, and optimize large language models; design and build LLM-based agent systems with tool use, planning, and multi-agent collaboration; optimize inference performance, GPU kernel optimization, and contribute to deployment of LLM-based products in real-world applications through collaboration with research and product teams.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
AMD
AMD
5 months ago

Senior GPU Kernel Optimization Engineer

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 16 hours agoStatus: Live

Job Summary

Join AMD as an LLM Training & Agent Engineer to train, fine-tune, and optimize large language models; design and build LLM-based agent systems with tool use, planning, and multi-agent collaboration; optimize inference performance, GPU kernel optimization, and contribute to deployment of LLM-based products in real-world applications through collaboration with research and product teams.
Location: Beijing
Workplace: Onsite
Employment Type: Full time
Job Function: Communications, PR & Community

Key Responsibilities

  • •Train, fine-tune, and optimize Large Language Models (LLMs), including but not limited to pretraining, SFT, and RLHF pipelines
  • •Design and develop LLM-based agent systems (e.g., tool use, planning and reasoning, multi-agent collaboration)
  • •Optimize LLM inference performance, including latency, throughput, and memory (VRAM) usage
  • •Participate in GPU computing optimization, including operator/kernel optimization and parallelization strategies
  • •Collaborate with research and product teams to drive the deployment of LLMs in real-world applications

Pay and Benefits

Salary: CNY 411,670 - 588,100 annually

Key Requirements

  • •Bachelor’s degree or above in Computer Science, Artificial Intelligence, or a related field
  • •4+ years of relevant development experience
  • •Proficient in at least one of Python or C++, with strong engineering skills
  • •Familiar with LLM training workflows, with hands-on experience in training or fine-tuning; experience deploying LLM-based products is a plus
  • •Experience in agent development (e.g., LangChain, in-house agents, tool use systems)
Experience:4+ yearsArtificial intelligenceMachine learning
Education:Bachelor's
Skills:CommunicationCollaborationProblem-solving
Languages:English
Tech Stack:PythonC++LangChainLLMRLHFCUDATritonMegatronDeepSpeedKV cacheGPU computing

Company Brief

AMD
Designs and produces semiconductor products including CPUs, GPUs, and adaptive SoCs for consumer, enterprise, and embedded markets, competing across PCs, data centers, and gaming industries.
Industry: Electronics Manufacturing
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: Santa Clara, United States
Founded: 1969
Glassdoor
Glassdoor: 3.9
WebsiteLinkedIn