PyTorch Integration Engineer (Python/C++)

Graphcore Early Careers; West of England; Hybrid; GBP 65,000 - 90,000; Full time

Benefits offered by this job

Flexible working: Balance your work

Generous leave: Take time to rest, 充

Retirement planning support: Up to 5%

Phantom equity: Share in Graphcore’s 0

Workplace experience: Office spaces, <

Peace of mind protection: Income and 0

Flexible benefits: Private medical, 0

Optional benefits: Dental cover, etc

Job summary

Graphcore is seeking a Software Engineer to join the PyTorch team and help make its hardware feel native inside ML frameworks. You will design, implement, optimize and support features across PyTorch, Triton and related technologies, enabling ML engineers to get more from Graphcore hardware without changing how they build models.

You will contribute to upstream PyTorch, support hybrid Python and C++ extensions, and collaborate with software, AI/ML, research and hardware teams.

Qualifications

• Proven software engineering experience and collaboration in teams.
• Experience developing Python libraries.
• Experience building performant C++ applications in industry.
• Understanding of linear algebra or numerical analysis.
• Proficient with Git and code reviews in a commercial setting.
• Willingness to explore large, unfamiliar open-source codebases independently.

Skills

Software engineering

Python libraries

Git

Open-source exploration

Numerical analysis

Collaborative teamwork

Tools

PyTorch

C++

Triton

Job description

Graphcore is seeking a Software Engineer to join the PyTorch team and help make its hardware feel native inside ML frameworks. You will design, implement, optimize and support features across PyTorch, Triton and related technologies, enabling ML engineers to get more from Graphcore hardware without changing how they build models.

You will contribute to upstream PyTorch, support hybrid Python and C++ extensions, and collaborate with software, AI/ML, research and hardware teams.