Apply on Employer Site

AMD · 10 hours ago

Staff Software Development Engineer - AI/ML Frameworks, C/C++/Python

San Jose, CA

Full-time

Onsite

Lead/Staff

$164K/yr - $246K/yr

AMD is a company focused on building innovative products that enhance computing experiences across various domains. The role involves providing technical leadership in the development of AI frameworks, optimizing deep learning frameworks for AMD GPUs, and collaborating with both internal teams and open-source maintainers.

Artificial Intelligence (AI)Cloud ComputingComputerEmbedded SystemsGPUHardwareSemiconductor

Growth Opportunities

No H1B

Hiring Manager

Tressa Cooper (she/her)

Responsibilities

Optimize Deep Learning Frameworks: Enhance and optimize frameworks like PyTorch, vLLM, SGLang for AMD GPUs in open-source repositories

Develop GPU Kernels: Create and optimize GPU kernels to maximize performance for specific AI operations

Develop & Optimize Models: Design and optimize deep learning models using quantization specifically for AMD GPU performance

Collaborate with GPU Library Teams: Work closely with internal teams to analyze and improve training and inference performance on AMD GPUs

Collaborate with Open-Source Maintainers: Engage with framework maintainers to ensure code changes are aligned with requirements and integrated upstream

Software Engineering Best Practices: Apply sound engineering principles to ensure robust, maintainable solutions

Qualification

AI/ML frameworksGPU Kernel DevelopmentC++ programmingPythonDeep Learning OptimizationProfilingBenchmarkingModel OptimizationCollaborationLeadershipTeam player

Required

Technical leadership in the development of various AI frameworks in the AMD ecosystem

Developing and optimizing deep learning frameworks for AMD GPUs

Engaging with both internal GPU library teams and open-source maintainers

Utilizing cutting-edge compiler technologies and advanced engineering principles

Working in a dynamic development environment

Excellent leadership and collaboration skills

Guiding other senior developers and domain experts

Optimizing Deep Learning Frameworks: Enhance and optimize frameworks like PyTorch, vLLM, SGLang for AMD GPUs in open-source repositories

Develop GPU Kernels: Create and optimize GPU kernels to maximize performance for specific AI operations

Develop & Optimize Models: Design and optimize deep learning models using quantization specifically for AMD GPU performance

Collaborate with GPU Library Teams: Work closely with internal teams to analyze and improve training and inference performance on AMD GPUs

Collaborate with Open-Source Maintainers: Engage with framework maintainers to ensure code changes are aligned with requirements and integrated upstream

Apply sound engineering principles to ensure robust, maintainable solutions

BS, MS or PhD in Computer Science, Computer Engineering, Electrical Engineering, or related technical fields