PyTorch & C++ Integration Engineer for AI Accelerators

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.