Software System Design Engineer - GPU Driver & Power Management

Advanced Micro Devices; Markham; Hybrid; CAD 75,000 - 100,000; Full time

Job summary

A leading technology company in York Region, Markham, is seeking a software engineer to contribute to machine learning research. In this role, you'll study and evaluate cutting-edge techniques in distributed training on AMD accelerators. Ideal candidates possess strong programming skills in Python and C/C++, and a relevant academic background. The position promotes career growth in a collaborative, hybrid environment.

Benefits

AMD benefits at a glance

Qualifications

• Prior software internship and/or machine learning research experience preferred.
• Proficient in Python and C/C++ programming.

Responsibilities

• Study techniques for distributed machine learning training and inference.
• Evaluate software and hardware implementation aspects of techniques.
• Identify proposed techniques for AMD’s HW/SW ecosystem.
• Support publishing of novel techniques in conferences/journals.

Skills

Technical problem solving

Programming skills (Python, C/C++)

Communication skills

Education

Bachelor’s, Master’s, or Ph.D. in Computer or Electrical Engineering

Tools

Machine learning frameworks

Job description

WHAT YOU DO AT AMD CHANGES EVERYTHING

At AMD, our mission is to build great products that accelerate next-generation computing experiences—from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you’ll discover the real differentiator is our culture. We push the limits of innovation to solve the world’s most important challenges—striving for execution excellence, while being direct, humble, collaborative, and inclusive of diverse perspectives. Join us as we shape the future of AI and beyond.

Together, we advance your career.

THE ROLE:

AMD is looking for a software engineer to join our Kernel Mode Driver and Power Management team and contribute to forward-looking machine learning research. Alongside senior machine learning researchers, you will work on evaluating and prototyping cutting‑edge solutions to implicit neural representations, and distributed training and inference techniques on next‑gen AMD accelerators. You will be encouraged to explore emerging techniques and support implementation of proof‑of‑concepts.

As a member of the Software Driver team you will work in conjunction with the Software Architecture team, you will be part of a group whose mandate is to research, prototype, and architect solutions for GPU compute & graphics and machine learning. Hence, you will get considerable support from experts in the various fields that are necessary to tackle interdisciplinary research topics surrounding distributed training and inference with deep neural networks.

THE PERSON:

Your curiosity will drive your learning and innovation to improve how we as a group, and an organization, can get better every day. Your peers will provide you a results‑oriented and encouraging environment for your career growth, fueling your opportunity to be a part of Delighting Our Customers.

KEY RESPONSIBILITIES:

• Study state‑of‑the‑art techniques for distributed machine learning training and inference, and implicit neural representations of images or videos.
• Evaluate software and hardware implementation aspects of these techniques and document findings.
• Identify/propose a set of techniques that are appropriate for AMD’s diverse HW/SW ecosystem.
• Support publishing non‑confidential aspects of novel technique(s) in tier‑1 conferences/journals.

PREFERRED EXPERIENCE:

• Candidate should be good at technical problem solving and have excellent programming skills, ideally with prior software internship and/or machine learning research experience.
• Experience with Python, C/C++, and machine learning frameworks.
• Excellent verbal and written communication skills.

ACADEMIC CREDENTIALS:

• Bachelor’s, Master’s, or Ph.D. degree in Computer or Electrical Engineering or equivalent (preferred).

#LI-TB1

#LI-Hybrid

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.