PMTS Software Development Engineer

AMD; Bellevue (WA); On-site; USD 120,000 - 180,000; Full time

Job summary

AMD is seeking a PMTS Software Development Engineer to develop and optimize low-level GPU kernels for accelerating inference and training of large ML models. You will work on distributed model training across multiple GPUs and nodes, profiling performance and optimizing resource utilization to maximize efficiency.

Collaborate with ML researchers and infrastructure teams to productionize optimized kernels and contribute to documentation and best practices.

Qualifications

• Five years of experience in the job offered or closely related engineering role.

Responsibilities

• Develop and optimize low-level GPU kernels to accelerate inference and training of large ML models.
• Profile and optimize system performance across GPUs and nodes to maximize throughput.
• Collaborate with ML researchers and software engineers to integrate optimized kernels into production systems.
• Create detailed documentation of optimizations and guidelines for maintainable code.

Skills

Low-level GPU kernels

Distributed ML training

Performance profiling

Verilog RTL coding

UVM

VLSI Design

Python scripting

TCL/TK

FPGA

Education

Master’s degree in CS/CE/EE/SE

Bachelor’s degree + 7 years experience

Tools

CUDA

Job description

ADVANCE YOUR CAREER. ADVANCE THE WORLD.

At AMD, we believe technology can change lives for the better. It can heal us, entertain us, and make us more connected, productive, and understanding of the world around us. And we’re looking for talent who feel the same: people who want to leave the planet better than they found it, those who don’t shy away from humanity’s challenges but are determined to help solve them.

AMD is powering the next generation of supercomputing, high-performance computing, cloud, and AI. Whether you’re designing next-gen processors, enabling AI breakthroughs, or creating go-to-market plans, every role at AMD contributes to something bigger — technology that moves the world forward.

Job Role and Responsibility:

AMD, Inc. is hiring PMTS Software Development Eng. to Develop and optimize low-level GPU kernels to accelerate inference and training of large machine learning models. Maximize the computational efficiency and reduce execution time while ensuring model accuracy. Design and implement strategies for distributed model training and inference across multiple GPUs and nodes. Address data parallelism and model parallelism challenges to fully utilize available resources. Profile and analyze system and application performance to identify bottlenecks and areas for improvement. Use profiling tools to understand and optimize hardware resource utilization. Leverage parallel computing techniques to improve the scalability and performance of machine learning workloads. Implement multi-threading and GPU synchronization techniques. Explore and apply model quantization techniques to reduce memory and computation overhead, especially for edge and cloud deployment. Develop benchmarks and testing procedures to assess the performance and stability of optimized models and frameworks. Ensure that the solutions meet or exceed the defined performance criteria. Collaborate closely with machine learning researchers, software engineers, and infrastructure teams to integrate optimized kernels and solutions into production systems. Create detailed documentation of optimizations, best practices, and implementation guidelines to facilitate knowledge sharing and maintainable code. Utilize knowledge of computers and electronics, including circuit boards, processors, chips, and electronic equipment, as well as knowledge of design techniques, tools, and principles. Apply knowledge of engineering principles, best practices, and technologies to the design, development, and testing of various company proprietary products.

Travel required:

NO

Qualifications: Degree required

Master’s degree or foreign equivalent in Computer Science, Computer Engineering, Electrical Engineering, Software Engineering, or related field.

Qualifications: Amount and type of experience required:

Five (5) years of experience in the job offered or closely related engineering role.

Alternate combination of education and experience:

Employer will accept alternate combination of education and experience: Bachelor’s degree or foreign equivalent in Computer Science, Computer Engineering, Electrical Engineering, Software Engineering, or related field and seven (7) years of progressive post-baccalaureate experience in the job offered or closely related engineering role.

Specific skills required: The following skills are required:

Position requires two (2) years of experience in the following:

• Designing, implementing, and testing computer hardware;
• Working with architecture and software teams to write specifications for the microarchitecture design;
• Power design concepts like multiple power domain, power gating, and data retention;
• Verilog RTL coding;
• UVM testbench;
• VLSI Design;
• Scripting languages, including Perl, Python, and TCL/TK; and
• FPGA.

Benefits offered are described: AMD benefits at a glance.

AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law. We encourage applications from all qualified candidates and will accommodate applicants’ needs under the respective laws throughout all stages of the recruitment and selection process.

AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMD’s “Responsible AI Policy” is available here.

This posting is for an existing vacancy.