Principal PyTorch Training Frameworks Engineer

Advanced Micro Devices; San Jose (CA); Hybrid; USD 210,000 - 290,000; Full time

Benefits offered by this job

Benefits at a glance

Job summary

A leading semiconductor company is seeking a Principal Software Development Engineer for their PyTorch Training Frameworks team. This role focuses on optimizing AI training performance on AMD hardware. Candidates should have a deep understanding of PyTorch internals, distributed training expertise, and strong programming skills in Python and C/C++. The ideal candidate will work closely with cross-functional teams to elevate the standard of PyTorch training functionalities, ensuring scalability and reliability. This position offers a hybrid work model with preferred locations in San Jose, Seattle, or Austin.

Qualifications

• Deep experience with PyTorch internals and training systems.
• Proven track record in performance engineering.
• Strong programming skills in Python and C/C++.

Responsibilities

• Act as a technical authority for PyTorch training at AMD.
• Drive optimization of key PyTorch training workloads.
• Contribute to and influence upstream PyTorch discussions.

Skills

Strong technical leadership

Hands-on with PyTorch training

Distributed training expertise

Programming skills in Python

Programming skills in C/C++

Clear technical communication

Education

Bachelor's or Master's in Computer Science or related field

Job description

A leading semiconductor company is seeking a Principal Software Development Engineer for their PyTorch Training Frameworks team. This role focuses on optimizing AI training performance on AMD hardware. Candidates should have a deep understanding of PyTorch internals, distributed training expertise, and strong programming skills in Python and C/C++. The ideal candidate will work closely with cross-functional teams to elevate the standard of PyTorch training functionalities, ensuring scalability and reliability. This position offers a hybrid work model with preferred locations in San Jose, Seattle, or Austin.