PyTorch Integration Engineer (Python/C++)

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.