PyTorch & C++ Integration Engineer for AI Accelerators

PyTorch & C++ Integration Engineer for AI Accelerators

Full-Time No working from home possible
G

Graphcore is seeking a Software Engineer to join the PyTorch team. You will help build software that connects Graphcore accelerators with state-of-the-art ML frameworks, designing, implementing, and optimizing features across PyTorch, Triton and related technologies.

You will contribute to upstream PyTorch, support hybrid Python and C++ extensions, and help integrate compiler backend capabilities with torch.compile.

#J-18808-Ljbffr

PyTorch & C++ Integration Engineer for AI Accelerators employer: graphcore

Graphcore is an exceptional employer, offering a vibrant work culture that fosters continuous learning and innovation in the heart of London. With a commitment to inclusivity and employee well-being, we provide flexible working arrangements, comprehensive benefits including private medical insurance and generous parental leave, and opportunities for professional growth within a cutting-edge AI environment. Join us to be part of a transformative journey in AI technology, where your contributions will directly impact the future of intelligent computing.

G

Contact Details:

graphcore Recruitment Team