Opensource Al workload Software Engineer

AMD
San Jose
Workplace: HybridFull timeFunction: Software EngineeringSkills: ["Communication","Teamwork","Problem-solving","Collaboration","Adaptability"]

Join AMD as a founding member of a core team to optimize AI workloads running on AMO GPUs. You will improve GPU performance for open-source LLM workloads (vLLM, SGLang), co-optimize workloads at the kernel level, and integrate AMO stacks (RoCm, ATen) with frameworks like PyTorch, JAX, and Triton while building strong cross-team partnerships.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
AMD
AMD
3 months ago

Opensource Al workload Software Engineer

✓ Verified Job

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

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

Job Summary

Join AMD as a founding member of a core team to optimize AI workloads running on AMO GPUs. You will improve GPU performance for open-source LLM workloads (vLLM, SGLang), co-optimize workloads at the kernel level, and integrate AMO stacks (RoCm, ATen) with frameworks like PyTorch, JAX, and Triton while building strong cross-team partnerships.
Location: San Jose
Workplace: Hybrid
Employment Type: Full time
Job Function: Software Engineering

Key Responsibilities

  • •Improve AMO GPU performance on open-source repositories for LLM workloads such as vLLM and SGLang.
  • •Co-optimize AI workloads on current AMO GPUs by analyzing bottlenecks and mitigating them at the kernel level.
  • •Integrate AMO software stacks (RoCm, ATen) into open-source frameworks such as PyTorch, JAX, and Triton.
  • •Build strong technical relationships with peers and partners, and report learnings and gaps to GPU software and hardware engineers.
  • •Collaborate to advance AI computing solutions and contribute to open-source ecosystems.

Key Requirements

  • •Hands-on experience with state-of-the-art LLM inference and training at both the framework and kernel levels
  • •Deep infrastructure experience with open-source frameworks (e.g., SGLang, vLLM, Jax, XLA, PyTorch, Triton)
  • •Strong kernel optimization skills using DSLs and HIP (or CUDA), plus PTX/SASS equivalents
  • •Demonstrated open-source contributions on GitHub
  • •Bachelor's, Master's, or Ph.D. in Computer Engineering, Computer Science, Electrical Engineering, or a related technical field
Experience:Open-sourceAIGPULLMInference
Skills:CommunicationTeamworkProblem-solvingCollaborationAdaptability
Languages:English
Tech Stack:PyTorchJAXTritonROCmATenVLLMSGLangHIPCUDAPTXSASSXLAGitHub

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