PyTorch Integration Engineer (Python/C++)

PyTorch Integration Engineer (Python/C++)

Full-Time On-site
G

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.

#J-18808-Ljbffr

PyTorch Integration Engineer (Python/C++) employer: Graphcore Early Careers

At Graphcore, we pride ourselves on being an exceptional employer, offering a dynamic work culture that fosters innovation and collaboration among our talented team of engineers. Located in the vibrant city of Bristol, we provide flexible working arrangements, generous benefits including private medical insurance and a supportive parental leave policy, and a commitment to employee growth through diverse project opportunities. Join us to make a meaningful impact in the fast-evolving field of AI compute, where your contributions will be valued and recognised.

G

Contact Details:

Graphcore Early Careers Recruitment Team