Principal PyTorch Training Frameworks Engineer

Advanced Micro Devices; San Jose (CA); Ibrido; USD 210.000 - 290.000; Tempo pieno

Vantaggi offerti da questo lavoro

Benefits at a glance

Descrizione del lavoro

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.

Competenze

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

Mansioni

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

Conoscenze

Strong technical leadership

Hands-on with PyTorch training

Distributed training expertise

Programming skills in Python

Programming skills in C/C++

Clear technical communication

Formazione

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

Descrizione del lavoro

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.